MAli-Farooq's picture
Update README.md
0df98a0 verified
metadata
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']