Ubuntu 16.04
CUDA 11.1
cuDNN 8.2.0
手電筒1.1.1+Cu 11
火炬視覺0.11.2+Cu 111
YOLOv5建立並訓練自己的數據集，請參考另壹篇文章:/p/b3fa74f6c27b。
訓練後會得到best.pt和last.pt的權重文件，在/run/train/exp/weights下。
以YOLOv5 v6.0為例。首先使用原始權重文件和項目提供的原始圖片進行detect推理，然後就可以修改detect.py文件了。
您可以先使用命令檢查命令是否正確。
然後，使用命令
運行結果將保存在/run/detect文件夾中。
檢測結果如下:
除了標記的圖片，檢測結果還有壹個文件夾記錄檢測框的信息:
妳可以根據以上修改自己，檢測自己的項目。
我在檢測的時候，沒有檢測盒，整天找bug。原來命令寫錯了，兩行淚傷心= =
壹定要按照python detect.py -h寫，不然不會對圖片做任何改動。