| 1234567891011121314151617181920212223 |
- 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]
|
