search

stefanopini / simple-HRNet

Multi-person Human Pose Estimation with HRNet in Pytorch
GNU General Public License v3.0
579 stars 108 forks source link

Using a different detector #94

Closed VivianaFigueira closed 2 years ago

VivianaFigueira commented 2 years ago

Hello, if i wanted to use a different detector, could i just switch the configuration and the weights, or is it more to it ? Without wanting to train the detector.

Thank you

stefanopini commented 2 years ago

Hi,

If it is just a different version of YOLO, you can try changing the configuration and the weights. It works for yolov3 and yolov3-tiny, but it may not work with different versions (e.g. #51 ).

If you want to use a completely different (pre-trained) detector, you can, but the output format must be the same of the current detector wrapper. For guidance, please take a look at the file models/detectors/YOLOv3.py.