Vscode configure les paramètres lors du débogage du train de modèles et détecte les scripts

Lorsque nous entraînons yolov5le code, il y a généralement 配置certains paramètres, tels que le fichier de poids du modèle --weights, le fichier de configuration du modèle --cfget les données d'entraînement --data.
Le script d'entraînement correspondant est :

entraînement

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

Paramètres des paramètres de débogage

Méthode 1 : définir les paramètres directement dans le code

Déboguer ensuite le code de train.py. Si vous ne définissez pas de paramètres, vous signalerez une erreur si vous déboguez directement. Une solution consiste à parse_optmodifier manuellement
--weights ， --cfg , --dataces trois paramètres dans la fonction, puis à définir un point d'arrêt et à appuyer sur F5pour déboguer. Évidemment, cette méthode nécessite de modifier manuellement le code, ce qui n'est pas très pratique. Puisque le test modifie les paramètres et les modifie la prochaine fois, il est facile d'oublier les paramètres d'origine.

Méthode 2 : configurer les paramètres dans launch.json

Cliquez sur Debugle bouton à droite pour sélectionner 创建launch.jsonle fichier.

Le code affiché à ce moment launch.jsonest le suivant :

{
    
    
    // 使用 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
        }
    ]
}

Dans , configurez les paramètres requis launch.jsonpour le débogage, ajoutez une nouvelle argsvariable, configurez --weights, et d'autres paramètres qui doivent être configurés.--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
        }
    ]
}

De cette façon, la configuration des paramètres d'entraînement est terminée, et vous pouvez casser le point et appuyer sur F5pour déboguer. Cette méthode sera plus pratique.