Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,11 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
---
|
| 4 |
+
The zip file contains all the YOLOV-5 variant models tuned on thermal data using two different SoA optimizers which includes ADAM and SGD. Each model variant contains two model versions which includes 'Best.pt' and 'Last.pt'. The models can also used in ensambling approach to increase the calss wise accuracy however it will take greater inference time.
|
| 5 |
+
|
| 6 |
+
train: ../train/images
|
| 7 |
+
val: ../valid/images
|
| 8 |
+
|
| 9 |
+
nc: 7
|
| 10 |
+
Classes names: ['bicycle', 'bike', 'bus', 'car', 'dog', 'person', 'pole']
|
| 11 |
+
|