search

amdegroot / ssd.pytorch

A PyTorch Implementation of Single Shot MultiBox Detector
MIT License
5.13k stars 1.74k forks source link

Where did you get the original Caffe weights file -- ssd_300_VOC0712.pth? #128

Closed xiaozhi2015 closed 6 years ago

xiaozhi2015 commented 6 years ago

@amdegroot What I got from original Caffe is VGG_VOC0712_SSD_300x300_iter_120000.caffemodel file. So where did you get the original Caffe weights file -- ssd_300_VOC0712.pth?

hengck23 commented 5 years ago

i verified that 'ssd_300_VOC0712.pth' is from 'VGG_VOC0712_SSD_300x300_iter_120000.caffemodel'. I made my own caffe to pytorch converter and have the exact same results for both files.

kirIn0305 commented 5 years ago

I want to convert .pth file from 'VGG_VOC0712_SSD_300x300_iter_120000.caffemodel. Do you change only the weights of first layer from rgb to bgr, and the other layers unchanged?

gjm441 commented 5 years ago

i verified that 'ssd_300_VOC0712.pth' is from 'VGG_VOC0712_SSD_300x300_iter_120000.caffemodel'. I made my own caffe to pytorch converter and have the exact same results for both files.

@hengck23 ,hi,can you share your scripts for converting ssd caffemodel to pth?Great thanks!