YOLOv5_Torchscript
YOLOv5_Torchscript copied to clipboard
zherlock030
C++ code for running a yolov5s model.
Tried to compilation YOLOv5_Torchscript: libtorch-cu101 cd YOLOv5_Torchscript mkdir -p build &&build cmake .. -- The C compiler identification is GNU 7.5.0 -- The CXX compiler identification is GNU 7.5.0 --...
Like what I said in the title. Many thanks for your great effort!
I converted the latest onnx and pt ,But it did not detect the target。can you give me export_onnx.py or .onnx
torchvision package doesn't looks like to be included in libtorch lib headers, from cMake file looks like `torchvision-abi/include/torchvision/nms.cpp` is a external module, it would be helpful if you can give...
