ObjectBox: From Centers to Boxes for Anchor-Free Object Detection

ECCV 2022 (Oral Presentation)

Dependencies

PyTorch

This code is tested under Ubuntu 18.04, CUDA 11.2, with one NVIDIA Titan RTX GPU.
Python 3.8.8 version is used for development.

Preparation

Set the 'PATH' in '/data/coco.yaml' and '/data/VOC.yaml'
Set the 'project' flag in flag_sets.py

Training

Set 'task' flag in flag_sets.py as: 'train'

For MS-COCO 2017 experiments, set:
exp = 'coco'
in flag_sets.py

For PASCAL VOC 2012 experiments, set:
exp = 'pascal'
in flag_sets.py

Run train.py

Test

Set 'task' flag in flag_sets.py as: 'test'

Run val.py

Acknowledgements

This project is supported by Geotab Inc., the City of Kingston, and the Natural Sciences and Engineering Research Council of Canada (NSERC)

Citation

Please cite our papers if you use code from this repository:

@article{zand2022objectbox,
  title={ObjectBox: From Centers to Boxes for Anchor-Free Object Detection},
  author={Zand, Mohsen and Etemad, Ali and Greenspan, Michael},
  journal={arXiv preprint arXiv:2207.06985},
  year={2022}
}

@article{zand2021oriented,
  title={Oriented bounding boxes for small and freely rotated objects},
  author={Zand, Mohsen and Etemad, Ali and Greenspan, Michael},
  journal={IEEE Transactions on Geoscience and Remote Sensing},
  volume={60},
  pages={1--15},
  year={2021},
  publisher={IEEE}
}

Reference

Many utility codes are borrowed from YOLO.

huyhoang17 / ObjectBox