评估代码在910上报hw_scale错误

[iamzjs]

我用的命令：python test.py --config ./configs/yolov5/yolov5n.yaml --weight ./yolov5n.ckpt \ --device_target Ascend

定位在这一行： File "/home/ma-user/work/mindyolo/mindyolo/data/dataset.py", line 1183, in test_collate_fn hw_scale = [sample.pop('hw_scale') for sample in batch_samples] 日志如下： Traceback (most recent call last): File "test.py", line 511, in main(args) File "test.py", line 502, in main synchronizer=Synchronizer(args.rank_size) if args.rank_size > 1 else None, File "test.py", line 121, in test return test_detect(**kwargs) File "test.py", line 159, in test_detect for i, data in enumerate(loader): File "/home/ma-user/anaconda3/envs/MindSpore/lib/python3.7/site-packages/mindspore/dataset/engine/iterators.py", line 145, in next data = self._get_next() File "/home/ma-user/anaconda3/envs/MindSpore/lib/python3.7/site-packages/mindspore/dataset/engine/iterators.py", line 270, in _get_next raise err File "/home/ma-user/anaconda3/envs/MindSpore/lib/python3.7/site-packages/mindspore/dataset/engine/iterators.py", line 253, in _get_next return {k: self._transform_md_to_output(t) for k, t in self._iterator.GetNextAsMap().items()} RuntimeError: Exception thrown from user defined Python function in dataset.

---

• Python Call Stack:

  KeyError: Traceback (most recent call last): File "/home/ma-user/anaconda3/envs/MindSpore/lib/python3.7/site-packages/mindspore/dataset/transforms/py_transforms_util.py", line 198, in call result = self.transform(*args) File "/home/ma-user/work/mindyolo/mindyolo/data/dataset.py", line 1183, in test_collate_fn hw_scale = [sample.pop('hw_scale') for sample in batch_samples] File "/home/ma-user/work/mindyolo/mindyolo/data/dataset.py", line 1183, in hw_scale = [sample.pop('hw_scale') for sample in batch_samples] KeyError: 'hw_scale'

---

• Dataset Pipeline Error Message:

  [ERROR] Execute user Python code failed, check 'Python Call Stack' above.

[zhanghuiyao]

这个看起来应该是数据处理问题，已在PR 257 中修复，可以更新代码试下