search

ShuangXieIrene / ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.
MIT License
566 stars 165 forks source link

YOLOv3 and Resnet #61

Closed robinenrico closed 5 years ago

robinenrico commented 5 years ago

I would like to train YOLO v3 with Resnet, does anyone has pre-trained weights for this configuration?

foreverYoungGitHub commented 5 years ago

You can use the resnet weight from imagenet or coco as pretrain weights. Cause the resnet occupies the majoir parts of the paramter, while the detection head only takes a small part.