Skip to content

Commit 8ec2da9

Browse files
authored
Yolo docs minimalist (#1723)
1 parent a5d007e commit 8ec2da9

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

docs/source/docs/objectDetection/about-object-detection.md

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,11 @@ Photonvision will letterbox your camera frame to 640x640. This means that if you
3535

3636
## Training Custom Models
3737

38-
Coming soon!
38+
:::{warning}
39+
Power users only. This requires some setup, such as obtaining your own dataset and installing various tools. It's additionally advised to have a general knowledge of ML before attempting to train your own model. Additionally, this is not officialy supported by Photonvision, and any problems that may arise are not attributable to Photonvision.
40+
:::
41+
42+
Before beginning, it is necessary to install the [rknn-toolkit2](https://github.com/airockchip/rknn-toolkit2). Then, install the relevant [Ultralytics repository](https://github.com/airockchip?tab=repositories&q=yolo&type=&language=&sort=) from this list. After training your model, export it to ``rknn``. This will give you an ``onnx`` file, formatted for conversion. Copy this file to the relevant folder in [rknn_model_zoo](https://github.com/airockchip/rknn_model_zoo), and use the conversion script located there to convert it. If necessary, modify the script to provide the path to your training database for quantization.
3943

4044
## Uploading Custom Models
4145

0 commit comments

Comments
 (0)