by pascal1129

Kaggle airbus ship detection challenge 21st solution

225 Stars 77 Forks Last release: almost 2 years ago (1.0.0) Apache License 2.0 12 Commits 2 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

Kaggle Airbus Ship Detection Challenge : 21st solution

This project is for Kaggle competiton Airbus Ship Detection Challenge.

It can help you quickly get a baseline solution, which is not bad.


Related article

These guides are only in Chinese:

Kaggle新手银牌(21st):Airbus Ship Detection 卫星图像分割检测

用Mask R-CNN训练自己的COCO数据集(Detectron)


File strcture

├─0_rle_to_coco                0、turn rle to coco
│  └─pycococreatortools
├─1_detectron_infer            1、files needed to be changed in detectron
|  ├─dataset_catalog.py            # ./detectron/datasets/dataset_catalog.py
│  ├─dummy_datasets.py             # ./detectron/datasets/dummy_datasets.py 
│  └─infer_airbus.py               # ./tools/infer_simple.py    
├─2_model                      2、model and trainning log
│  ├─log                           log and visualization script
│  └─model                         configure file and .pkl (.pkl not be uploaded)
└─3_submit                     3、generate your submission
   └─csv                           reference .csv file


1. Generate COCO standard dataset

Run codes in

. The guide has been written in markdwon file

dataset annotation

2. Get Detectron environment

My codes are based on Detectron. So before using it, you need to install caffe2, which is quite troublesome. You can use my docker image, which is a little out of date, by the following command:

$ docker pull pascal1129/detectron:caffe2_cuda9_aliyun

In order to get the latest docker image, you can build the latest image with the official dockerfile: Detectron/docker/Dockerfile.

3. Msodify the source code in detectron

My codes are in the folder

, you can replace the origin files in detectron with my codes.

my code

origin code needed to be replaced
datasetcatalog.py ./detectron/datasets/datasetcatalog.py
dummydatasets.py ./detectron/datasets/dummydatasets.py
inferairbus.py ./tools/infersimple.py

4. Change the configuration file and run

Confirm the .yaml file in

and start training. In addition, remember to use
command, so you can get the log file like ./2_model/log/20181103.log

5. Visualization


, then you can get the visualization picture.


6. Get the final submission



We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.