MMDetection Baseline

#10
by rashmi - opened

Baseline with MMDetection

MMDetection training code

Inference

You can find coco format conversion here in cell [10] - Ship Detection EDA

I ran both training and inference locally - so version 1 won't run on kaggle kernel. (version 2 is running on kaggle) You can try to upload dataset and adjust paths for running.
(Last time I tried installing in kernel, it took 30 mins, probably I had incorrect version)

Some ideas to improve -

  • See model zoo here
  • Hyperparameters
  • Threshold
  • I used 80/20 split for validation, make it 90/10 or higher, 5 folds - ensembles etc

Any bugs/suggestions/questions please let me know!

I adjusted versions for training code and Inference - so now they run on kaggle ๐ŸŽ‰

Sign up or log in to comment