Add YOLOv6 support

2023-02-01 02:52:01 -03:00
parent 69f29f8934
commit 087a41acf6
19 changed files with 982 additions and 65 deletions
--- a/docs/YOLOv5.md
+++ b/docs/YOLOv5.md
@@ -46,6 +46,12 @@ Generate the `cfg` and `wts` files (example for YOLOv5s)
 python3 gen_wts_yoloV5.py -w yolov5s.pt
 ```

+**NOTE**: To convert a P6 model
+
+```
+--p6
+```
+
 **NOTE**: To change the inference size (defaut: 640)

 ```