strides = [8, 16, 32]

# fire
# model_type = 'yolov8'
# net_shape=(384, 640)
# names = ['fire']
# cls_num = len(names)
# image_base_path = '/root/data_ssd/fire/dataset/images/test'
# label_base_path = '/root/data_ssd/fire/dataset/labels/test'
# bin_path = ['/sim/res_hw/_model.22_Concat.bin', '/sim/res_hw/_model.22_Concat_1.bin', '/sim/res_hw/_model.22_Concat_2.bin']
# output_names = ['/model.22/Concat', '/model.22/Concat_1', '/model.22/Concat_2']
# fl = [2, 2, 2]

# baby head
model_type = 'yolov8'
net_shape=(384, 640)
names = ['head']
cls_num = len(names)
image_base_path = '/root/data_ssd/dataset_baby_head/images/test'
label_base_path = '/root/data_ssd/dataset_baby_head/labels/test'
bin_path = ['/sim/res_hw/_model.22_Concat.bin', '/sim/res_hw/_model.22_Concat_1.bin', '/sim/res_hw/_model.22_Concat_2.bin']
output_names = ['/model.22/Concat', '/model.22/Concat_1', '/model.22/Concat_2']
fl = [2, 2, 2]