问题描述

gitee.com/mindspore的model仓里的r2.1及其master分支的yolov5代码的yolo.py文件代码有问题。
在第356行——359行代码会导致报错。
源代码：
本人尝试修改后的代码：

修改后是可以跑通的。建议后端的专家复现并修改上传的代码，少点bug，以保证一线能正常跑通。

环境信息

• Hardware Environment(Ascend/GPU/CPU) / 硬件环境: Ascend910B1

Please delete the backend not involved / 请删除不涉及的后端:
/device Ascend

• Software Environment / 软件环境 (Mandatory / 必填):cann_6.3.2
  -- MindSpore version (e.g., 2.0.0) :mindspore_2.1.0
  -- Python version (e.g., Python 3.7.5) :py_3.7
  -- OS platform and distribution (e.g., Linux Ubuntu 16.04):euler_2.10.7-aarch64

• Execute Mode / 执行模式 (Mandatory / 必填)(PyNative/Graph): ms.GRAPH_MODE

Please delete the mode not involved / 请删除不涉及的模式:
/mode graph

关联用例

报错截图：
修改后跑通截图：

重现步骤

1、下载gitee.com/mindspore的models仓r2.1分支的yolov5代码
2、准备coco2017数据集
3、执行bash run_standalone_train.sh [DATASET_PATH] [DEVICE_ID]
指定数据集路径和device id（比如0）
4、查看train/log.txt

预期结果

代码bug修复，一线按照文档下载代码和数据集，上传后，执行脚本可以开箱即可跑通。

日志/截图

报错日志：

备注