YOLOv5_Torchscript
YOLOv5_Torchscript copied to clipboard
Q: Does this model support batch dimension?
Like what I said in the title. Many thanks for your great effort!
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.