wangjin0928 68b7ca8abc add forest2 | před 1 rokem | |
---|---|---|
weights | před 1 rokem | |
readme.md | před 1 rokem |
所有的weights下的文件夹是按业务类型组织 一、每一个业务文件夹内容如下 labelnames.json -- 类别配置文件 para.json -- 检测相关参数 yolov5.pt -- 模型原始权重,一般用不到 yolov5_2080Ti_fp16.engine --- 2080Ti下的TRT加速模型 yolov5_3090_fp16.engine --- 3090下的TRT加速模型 yolov5_A10_fp16.engine --- A10下的TRT加速模型
二、每种显卡每个文件夹下面以下配置文件是必须的 labelnames.json para.json 三、每种显卡下面一个.engine是必须的
3.1 如2080Ti显卡下面,以下文件是必须的
forest/yolov5_2080Ti_fp16.engine
river/stdc_360X640_2080Ti_fp16.engine
road/yolov5_2080Ti_fp16.engine
vehicle/yolov5_2080Ti_fp16.engine
pedestrian/yolov5_2080Ti_fp16.engine
river/yolov5_2080Ti_fp16.engine
smogfire/yolov5_2080Ti_fp16.engine
3.2 如3090显卡下面,以下文件是必须的
forest/yolov5_3090_fp16.engine
river/stdc_360X640_3090_fp16.engine
road/stdc_3090_fp16.engine
smogfire/yolov5_3090_fp16.engine
pedestrian/yolov5_3090_fp16.engine
river/yolov5_3090_fp16.engine
road/yolov5_3090_fp16.engine
vehicle/yolov5_3090_fp16.engine
3.3 如A10显卡下面,以下文件是必须的
forest/yolov5_A10_fp16.engine
river/stdc_360X640_A10_fp16.engine
road/stdc_A10_fp16.engine
smogfire/yolov5_A10_fp16.engine
pedestrian/yolov5_A10_fp16.engine
river/yolov5_A10_fp16.engine
road/yolov5_A10_fp16.engine
vehicle/yolov5_A10_fp16.engine