REVA-QCAV/.vscode/launch.json

{
  "version": "0.2.0",
  "configurations": [
    {
      "name": "Train",
      "type": "python",
      "request": "launch",
      "program": "${workspaceFolder}/src/train.py",
      "console": "integratedTerminal",
      "justMyCode": false,
    },
    {
      "name": "Predict",
      "type": "python",
      "request": "launch",
      "program": "${workspaceFolder}/src/predict.py",
      "console": "integratedTerminal",
      "justMyCode": false,
      "args": [
        "--input",
        "images/input.png",
        "--output",
        "images/output.png",
        "--model",
        "checkpoints/model.onnx"
      ]
    }
  ]
}