MAli-Farooq's picture
Update README.md
0df98a0 verified
---
license: mit
---
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 class wise accuracy however it will take greater inference time.
train: ../train/images
val: ../valid/images
nc: 7
Classes names: ['bicycle', 'bike', 'bus', 'car', 'dog', 'person', 'pole']