TNTWEN / OpenVINO-YOLOV4

This is implementation of YOLOv4,YOLOv4-relu,YOLOv4-tiny,YOLOv4-tiny-3l,Scaled-YOLOv4 and INT8 Quantization in OpenVINO2021.3
MIT License
238 stars 66 forks source link

where to get the .xml model? #2

Closed dsp6414 closed 4 years ago

TNTWEN commented 4 years ago

Now YOLOV4,V4tiny,V4relu are all supported. If you encounter other problems while using it, you can continue to contact me

dsp6414 commented 4 years ago

thanks

xaerincl commented 4 years ago

Does it matters if i trained my tiny-yolov4 with a different width and height than the standard 416x416 ?

TNTWEN commented 4 years ago

Hi,@xaerincl You could set the width and height you want. The default size is 416*416: https://github.com/TNTWEN/OpenVINO-YOLOV4/blob/3f5355f9c9ac9b9e2996457f4cdf9f2a40675c34/convert_weights_pb.py#L25

For example,you could use python convert_weights_pb.py --class_names cfg/coco.names --weights_file yolov4.weights --data_format NHWC --size 608 to set 608*608 size. You just need to set it to match the length and width of the model you trained. Under normal circumstances, we use the same length and width(416*416,608*608,832*832),if you trained a rectangular training model,you could try to change the code yourself.

xaerincl commented 4 years ago

nice, thanks!