部署yolov5代码之后报错
收藏回复举报
部署yolov5代码之后报错
t('forum.solved') 已解决
发表于2024-05-29 11:21:44
0 查看
在官方样例代码中更改代码中 

cfg = { 

    'conf_thres': 0.4,  # 模型置信度阈值,阈值越低,得到的预测框越多 

    'iou_thres': 0.5,  # IOU阈值,高于这个阈值的重叠预测框会被过滤掉 

    'input_shape': [640, 640],  # 模型输入尺寸 

 

model_path = 'yolo.om' 

label_path = './coco_names.txt' 

# 初始化推理模型 

model = InferSession(0, model_path) 

labels_dict = get_labels_from_txt(label_path) 

为 

cfg = { 

    'conf_thres': 0.4,  # 模型置信度阈值,阈值越低,得到的预测框越多 

    'iou_thres': 0.5,  # IOU阈值,高于这个阈值的重叠预测框会被过滤掉 

    'input_shape': [640, 640],  # 模型输入尺寸 

 

model_path = 'yolov5s2.om' 

label_path = './coco_names.txt' 

# 初始化推理模型 

model = InferSession(0, model_path) 

labels_dict = get_labels_from_txt(label_path) 

运行后报错,并且报错信息一直重复不停,只能强制停止

开发板为Atlas 200I DK A2,芯片为310B4%E5%BE%AE%E4%BF%A1%E5%9B%BE%E7%89%87_20240529111637.png

我要发帖子