Replacing CenterNet detector with custom trained detector

xingyizhou / GTR

Global Tracking Transformers, CVPR 2022

375 stars 57 forks source link

Replacing CenterNet detector with custom trained detector #19

Open InfiniteLife opened 2 years ago

InfiniteLife commented 2 years ago

Is there a way to replace CenterNet detector with Yolov5? Apparently your model build deep into Detectron2 package and Im not sure how to go about replacing detector.