当前位置:   article > 正文

解决:pycocotools unable to run: module ‘numpy‘ has no attribute ‘float‘. `np.float_coco128数据集txt格式转json格式

coco128数据集txt格式转json格式

1.原因:数据集中的标签与图像名称不匹配,有缺失。

2. 数据集:coco128

3. 场景:在使用yolo2coco.py脚本中,将coco128数据集由txt文件转为json文件。

4. 目的:在yolov5或yolov7中为了得到coco指标。

5. coco128数据集格式:

coco128

        images

                train2017

                        01.jpg

                        02.jpg

        labels

                train2017

                        01.txt

                        02.txt

6. 解决:运行代码,根据输出结果,找出images文件与labels文件中不匹配的文件进行删除。

  1. import os
  2. def split_str(lists):
  3. d = list()
  4. for s in lists:
  5. d.append(s.split('.')[0])
  6. return d
  7. path_img = './coco128/images/train2017'
  8. datanames_img = os.listdir(path_img) # 存放在列表中 '000000000009.jpg'
  9. list_img = split_str(datanames_img)
  10. path_labels = './coco128/labels/train2017'
  11. datanames_labels = os.listdir(path_labels)
  12. list_labels = split_str(datanames_labels)
  13. print('图片中存在,但标签中不存在')
  14. for i in list_img:
  15. if i not in list_labels:
  16. print(i)
  17. print('========')
  18. print('图片中不存在,标签中存在')
  19. for i in list_labels:
  20. if i not in list_img:
  21. print(i)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/不正经/article/detail/275833
推荐阅读
相关标签
  

闽ICP备14008679号