avBuffer / Yolov5_tf

Yolov5/Yolov4/ Yolov3/ Yolo_tiny in tensorflow
288 stars 108 forks source link

yolov5.pt #2

Open pioneer158 opened 4 years ago

pioneer158 commented 4 years ago

How to use Yolov5 weight file for detection?

avBuffer commented 4 years ago

Hey, Step1: do scripts/from_darknet_weights_to_pb.py to change into pb file, if yolov5.weights. However if yolov5.pt in Pytorch, it's a bit difficult. You should change pt into onnx, and then change into tensorflow pb

pioneer158 commented 4 years ago

thank you for your answer!  I have another problem ,which I use yolov4.pt to detect a image,but its performance is not good.  The detection result in accessory

------------------ 原始邮件 ------------------ 发件人: "Jay Meng"<notifications@github.com>; 发送时间: 2020年7月6日(星期一) 下午5:07 收件人: "avBuffer/Yolov5_tf"<Yolov5_tf@noreply.github.com>; 抄送: "调得一手好参"<928343544@qq.com>; "Author"<author@noreply.github.com>; 主题: Re: [avBuffer/Yolov5_tf] yolov5.pt (#2)

Hey, Step1: do scripts/from_darknet_weights_to_pb.py to change into pb file, if yolov5.weights. However if yolov5.pt in Pytorch, it's a bit difficult. You should change pt into onnx, and then change into tensorflow pb

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.