Closed bogdan-evtushenko closed 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
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.
@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 =)
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 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?
@bogdan-evtushenko yeeep, are you in the discord server already?
This is done.
🚀 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: