OlafenwaMoses / ImageAI

A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities
https://www.genxr.co/#products
MIT License
8.5k stars 2.18k forks source link

COCO annotation support #658

Open kegsay opened 3 years ago

kegsay commented 3 years ago

I noticed in the documentation that ImageAI uses Pascal VOC for annotations. Is there support (or planned support) for COCO annotations? I have an existing dataset using COCO annotations which contain more specific labelling data (polygons rather than just bounding boxes) and it looks like the VOC parser only uses bounding boxes.

Thanks, and fantastic project by the way!

ekesdf commented 3 years ago

Yea COCO is supported Just use it like a VOC Dataset.

But imageai just supports bounding boxes

But you can train a model using the tensorflow 2 object detection api [https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/]()