当前位置:   article > 正文

KITTI-MOT图片形式的Annotation解码成一般形式

kitti-mot
  1. from PIL import Image
  2. import glob
  3. import os
  4. # 对图片解码
  5. def decode_mask(path):
  6. out_base_path = '...../KITTI-tracking/decode_mask' #输出文件夹
  7. folders = sorted(glob.glob(os.path.join(path, '*')))
  8. for folder in folders:
  9. temp_1 = folder.split('/')
  10. samples = []
  11. samples.extend(sorted(glob.glob(os.path.join(folder, '*.png'))))
  12. out_path = os.path.join(out_base_path, '{}'.format(temp_1[-1]))
  13. os.makedirs(out_path, exist_ok=True)
  14. for sample in samples:
  15. temp_2 = sample.split('/')
  16. mask = Image.open(sample)
  17. mask_8bits = mask.convert('L')
  18. out = os.path.join(out_path, temp_2[-1])
  19. mask_8bits.save(os.path.join(out_path,out))
  20. decode_mask(#路径#)

                                                           

 

 

 

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

闽ICP备14008679号