Skip to content

Commit 36e787f

Browse files
YOLOv5: Explain how to use your own model and update links (#429)
Co-authored-by: Isak Jakobsson <[email protected]>
1 parent 17b9924 commit 36e787f

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

object-detection-yolov5/README.md

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -118,12 +118,11 @@ This example uses a YOLOv5n model trained on the [COCO dataset](https://cocodata
118118
on 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)
120120
or [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

Comments
 (0)