search

z393 / yolov3-tensorrt

pytorch-caffe-tensorrt
1 stars 0 forks source link

没有pytorch转caffe的相关代码？ #1

Open CangHaiQingYue opened 4 years ago

CangHaiQingYue commented 4 years ago

您好，工程中没有看到pytorch转caffe的相关代码，如models文件夹找不到。 是不是还没有上传。。。

z393 commented 4 years ago

您好，我这里用的是将 pytorch训练出来的模型， 利用yolov3-pytorch/model.py 将.pt文件转成.weights文件，然后用darknet2caffe/darknet2caffe.py 将.weights文件转为caffemodel.　希望可以帮助到你 :)

CangHaiQingYue commented 4 years ago

谢谢回复。 还有个问题请教下，再利用darknet2caffe.py将.weights文件转为caffemodel时，需要.cfg文件么？

z393 commented 4 years ago

需要的. 另外darknet2caffe时需要在caffe中加入upsample层并重新编译caffe, 我在darknet2caffe这个目录下写了readme, 您可以看一下 :)

CangHaiQingYue commented 4 years ago

emmm,我这边的一个yolov3没有cfg文件，可能比较麻烦~https://github.com/ouyanghuiyu/yolo-face-with-landmark