下載訓練模型:

github.com/ultralytics/yolov5/releases

*.pt檔即是訓練模型

 

 

下載後將*.pt檔放入yolov5資料夾內

 

安裝yolov5必要元件,

進入yolov5目錄下

輸入 pip install -r requirements.txt

 

圖片測試

透過detect.py對圖片進行物件檢測

 python detect.py --source ./data/images/bus.jpg

 

檢查輸出,

在yolov5資料夾內,\runs\detect\exp

 

 

到這表示物件偵測,AI模組成功安裝

 

 

 

創作者介紹
創作者 IMM 的頭像
IMM-AT

IMM

IMM-AT 發表在 痞客邦 留言(0) 人氣( 3 )