/augmented_detr

Implementation for Performance Improvement of Transformer Based Models Using Data Augmentations in Object Detection Task

Primary LanguageJupyter NotebookMIT LicenseMIT

augmented_detr

Implementation for Performance Improvement of Transformer Based Models Using Data Augmentations in Object Detection Task

MS C++ Build tools required. Follow readme for installing pycoco: https://github.com/philferriere/cocoapi

Data:

  • Download coco and voc data and place inside notebooks folder src notebooks data coco images train val test annotations train val test voc train images test images