@@ -118,12 +118,11 @@ This example uses a YOLOv5n model trained on the [COCO dataset](https://cocodata
118118on which ` <CHIP> ` is provided when [ building the application] ( #build-the-application ) , either
119119[ yolov5n_artpec8_coco_640.tflite] ( https://acap-ml-models.s3.us-east-1.amazonaws.com/yolov5/yolov5n_artpec8_coco_640.tflite )
120120or [ yolov5n_artpec9_coco_640.tflite] ( https://acap-ml-models.s3.us-east-1.amazonaws.com/yolov5/yolov5n_artpec9_coco_640.tflite )
121- is used. The differences between these models are explained in their respective training guide:
122-
123- - [ YOLOv5 on ARTPEC-8] ( https://github.com/AxisCommunications/axis-model-zoo/blob/main/docs/yolov5-on-artpec8.md ) .
124- - [ YOLOv5 on ARTPEC-9] ( https://github.com/AxisCommunications/axis-model-zoo/blob/main/docs/yolov5-on-artpec9.md ) .
125-
126- To train your own YOLOv5 model, follow the appropriate guide above.
121+ is used. The differences between these models are explained in the
122+ [ YOLOv5 training guide] ( https://github.com/AxisCommunications/axis-model-zoo/blob/main/docs/yolov5.md ) .
123+ That guide can also be used to train your own YOLOv5 model. Then, see the
124+ [ Dockerfile parameters] ( #dockerfile-parameters ) section to learn how to use your own model in this
125+ example.
127126
128127## Explanation of YOLOv5 output parsing
129128
@@ -219,8 +218,12 @@ installed. In order to apply the changes, the ACAP application must be restarted
219218
220219### Dockerfile parameters
221220
222- The model ` .tflite ` file and the labels ` .txt ` file are provided through the ` Dockerfile ` . In this
223- example, these files are downloaded using the cURL command.
221+ The model ` .tflite ` file and the labels ` .txt ` file are provided through the
222+ [ Dockerfile] ( ./Dockerfile ) . In this example, these files are downloaded using the ` curl ` command.
223+
224+ To use your own YOLOv5 model and labels file, you just have to configure the
225+ [ Dockerfile] ( ./Dockerfile ) to download your files, or, if you have the files locally, use the ` COPY `
226+ command instead.
224227
225228### Model-specific parameters
226229
0 commit comments