/EfficientDet

EfficientDet for custom dataset detection.

Primary LanguagePythonApache License 2.0Apache-2.0

EfficientDet

The EfficientDet and Backbone are from the amazing repo of rwightman. Check his repo to learn more about other amazing architectures.

Special thanks to Alex Shonenkov for providing great kernel. Don't forget to upvote his work :3.

Training

Training EfficientDet is a painful and time-consuming task.

STEP 1 -

# your dataset structure should be like this
datasets/
    -your_project_name/
        -train_set_name/
            -*.jpg
        -val_set_name/
            -*.jpg
        -annotations
            -instances_{train_set_name}.json
            -instances_{val_set_name}.json

# for example, coco2017
datasets/
    -coco2017/
        -train2017/
            -000000000001.jpg
            -000000000002.jpg
            -000000000003.jpg
        -val2017/
            -000000000004.jpg
            -000000000005.jpg
            -000000000006.jpg
        -annotations
            -instances_train2017.json
            -instances_val2017.json

STEP 2-

python main.py -m train

Coming Soon!