search

zherlock030 / YOLOv5_Torchscript

C++ code for running a yolov5s model.
28 stars 5 forks source link

Q: Does this model support batch dimension? #4

Open ganler opened 4 years ago

ganler commented 4 years ago

Like what I said in the title. Many thanks for your great effort!

zherlock030 commented 4 years ago

My implementation is according to detect.py in https://github.com/ultralytics/yolov5/blob/master/detect.py. I'll test batching and reply in days.