Vscode は、モデルのトレーニングと検出スクリプトのデバッグ時にパラメーターを構成します

yolov5コードをトレーニングするときは、通常、配置モデルの重みファイル--weights、モデル構成ファイル--cfg、トレーニング データなどのパラメーターがいくつかあります--data。
対応するトレーニング スクリプトは次のとおりです。

トレーニング

python train.py   -- weights './yolov5s.pt' --cfg 'models\yolov5s.yaml' --data './data/coco128.yaml'

デバッグパラメータ設定

方法 1: コード内でパラメータを直接設定する

次に、train.py のコードをデバッグします。パラメータを設定せずに直接デバッグすると、エラーが報告されます。1 つの方法は、関数内のこれら 3 つのパラメーターを手動でparse_opt変更し、ブレークポイントを設定して、 を押してデバッグすることです。この方法では、コードを手動で変更する必要があるのは明らかですが、あまり便利ではなく、テストではパラメータを変更し、次回は元に戻すため、元のパラメータ設定を忘れがちです。
--weights ， --cfg , --dataF5

方法 2: launch.json でパラメーターを構成する

Debug右側のボタンをクリックして创建launch.jsonファイルを選択します。

この時に表示されるコードはlaunch.json以下の通りです。

{
    
    
    // 使用 IntelliSense 了解相关属性。 
    // 悬停以查看现有属性的描述。
    // 欲了解更多信息，请访问: https://go.microsoft.com/fwlink/?linkid=830387
    "version": "0.2.0",
    "configurations": [
        {
    
    
            "name": "Python: Current File",
            "type": "python",
            "request": "launch",
            "program": "${file}",
            "console": "integratedTerminal",
            "justMyCode": true
        }
    ]
}

でlaunch.json、デバッグに必要なパラメータを設定し、新しいargs変数を追加し、configure --weights、--data、--cfgおよび設定が必要なその他のパラメータを追加します。

{
    
    
   
    "version": "0.2.0",
    "configurations": [
        {
    
    
            "name": "Python: Current File",
            "type": "python",
            "request": "launch",
            "program": "${file}",
            "args": [
            	"--weights", "./yolov5s.pt",
            	"--data", "./data/coco128.yaml",
            	"--cfg", "models\yolov5s.yaml",
            ],
            "console": "integratedTerminal",
            "justMyCode": true
        }
    ]
}

このようにして、トレーニングパラメータの設定が完了し、ポイントをブレークして を押してF5デバッグすることができるので、この方法の方が便利です。