airctic / icevision

An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come
https://airctic.github.io/icevision/
Apache License 2.0
842 stars 150 forks source link

Add state of the art YOLOX model from YOLO series. #911

Closed bogdan-evtushenko closed 2 years ago

bogdan-evtushenko commented 2 years ago

🚀 Feature

Is your feature request related to a problem? Please describe. The problem is basically in not sufficiency of mAP score for not big dataset size available.

Describe the solution you'd like Adding model the newest version from YOLO series - YOLOX - to the list of available models to use (github) (arxiv)

Describe alternatives you've considered We did a couple of experiments and that's what we had:

roboserg commented 2 years ago

Why YOLOX, and not YOLOR? YOLOR has better COCO AP over YOLOX - https://miro.medium.com/max/875/0*GHtVCZCRiHWhaQfI

YOLOX is even worse then Scaled-YOLOv4

bogdan-evtushenko commented 2 years ago

It's true, so from what I see from best to worse (COCO AP)(@roboserg correct me if I'm wrong): YOLOR -> Scaled-YOLOv4 -> YOLOX -> YOLOv5 So adding any of those could improve icevision usage experience.

lgvaz commented 2 years ago

@bogdan-evtushenko or @roboserg would you be up to adding this? It's a lot of work but should be straight forward, it should follow the same structure as our other models. If you're interested we can have a more detailed chat =)

weiji14 commented 2 years ago

YOLOX is available in mmdetection (https://github.com/open-mmlab/mmdetection/tree/v2.16.0/configs/yolox), so might be easier to integrate into icevision at https://github.com/airctic/icevision/tree/master/icevision/models/mmdet/models :wink:

bogdan-evtushenko commented 2 years ago

@bogdan-evtushenko or @roboserg would you be up to adding this? It's a lot of work but should be straight forward, it should follow the same structure as our other models. If you're interested we can have a more detailed chat =)

Yeap, I'm interested in adding that, let's have a chat?

lgvaz commented 2 years ago

@bogdan-evtushenko yeeep, are you in the discord server already?

FraPochetti commented 2 years ago

This is done.