object-detection-yolov5.json

{
  "plugins": {
    "dataCollect": {
      "package": "@pipcook/plugins-object-detection-coco-data-collect",
      "params": {
        "url": "http://ai-sample.oss-cn-hangzhou.aliyuncs.com/image_classification/datasets/autoLayoutGroupRecognition.zip"
      }
    },
    "dataAccess": {
      "package": "@pipcook/plugins-coco-data-access"
    },
    "modelDefine": {
      "package": "@pipcook/plugins-pytorch-yolov5-model-define"
    },
    "modelTrain": {
      "package": "@pipcook/plugins-pytorch-yolov5-model-train",
      "params": {
        "epochs": 100
      }
    },
    "modelEvaluate": {
      "package": "@pipcook/plugins-pytorch-yolov5-model-evaluate"
    }
  }
}