判断图片是否损坏的代码

有时候在用caffe训练模型的时候,会因为数据集的某些图片损坏而出现一些莫名其妙的错误,因此写了这样一个判断图片是否可用的代码,用来清洗数据

   def get_image_files():
       path = 'images_path'
      files = os.listdir(path)
      s=[]
      for file in files:
          s.append(file)
      return s
  def remove_bad_images():
      s = get_image_files()
     for image in s:
          try:
              image_file = 'images_path' +image
              img = Image.open(image_file)
              img.verify()
          except IOError:
              print(image_file)
             shutil.move(image_file,'/home/data/errimg')
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容