yolov5s:
從github官網下載yolov5的算法之后,配置好環境(pycharm安裝包-CSDN博客),再下載權重文件,比如默認的yolov5s.pt;
運行當前文件(detect.py),就能看到exp文件夾中的結果
也可以自定義參數運行:
?train.py和detect.py文件同時使用(訓練自己的數據集)
先運行train.py(上面一行命令) 將其中的data.yaml(可自定義名字)中訓練、驗證的路徑改成自己的數據庫文件夾,權重文件看要求可大可小;
訓練完之后生成best.pt和last.pt文件,
運行下面一行,導入檢測圖像和剛剛訓練好的best.pt文件即可