search

frgfm / Holocron

PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Other additions: AdEMAMix
https://frgfm.github.io/Holocron/latest
Apache License 2.0
313 stars 49 forks source link

Pretrained object detection #253

Open ShivaThe opened 1 year ago

ShivaThe commented 1 year ago

is there a pretrained model available for object detection. I tried using yolov4 but getting error in loading pretrained weight as there is no url in the script.

frgfm commented 1 year ago

Hello @ShivaThe :wave:

You are correct, I haven't provided a set of trained params for object detection models so far! I've been running a bit behind my schedule, but I'll do my best to provide some shortly.

If you have some computing power though, you can easily train it on your own using the reference script! This is already referenced in #110, so if that's alright, perhaps we can continue the discussion over there and close this issue :) Let me know if I misunderstood your request!