您当前的位置: 首页 > 
  • 0浏览

    0关注

    2393博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

CV之FE:基于TF进行FE——去除异常(被损坏)图像 和单通道图像

一个处女座的程序猿 发布时间:2018-10-08 20:30:40 ,浏览量:0

CV之FE:基于TF框架对图像进行数据预处理—去除异常(被损坏)图像 和单通道图像代码实现

目录

基于TF框架对图像进行数据预处理—去除异常(被损坏)图像 和单通道图像代码实现

输出结果

设计思路

部分代码实现

基于TF框架对图像进行数据预处理—去除异常(被损坏)图像 和单通道图像代码实现 输出结果

去除了异常(被损坏)图像 、单通道图像

设计思路

1、

部分代码实现
import tensorflow as tf
from glob import glob
import os

……


def glob_all(dir_path):
    pic_list = glob(os.path.join(dir_path, '*.jpg'))
    inside = os.listdir(dir_path)
    for dir_name in inside:
        if os.path.isdir(os.path.join(dir_path, dir_name)):
            pic_list.extend(glob_all(os.path.join(dir_path, dir_name)))
    return pic_list


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('-p', '--dir-path', default='data/')
    return parser.parse_args()

if __name__ == '__main__':
    #1、引入logging模块来记录信息,日志级别为INFO
    logging.basicConfig(level=logging.INFO)
    #2、引入argparse模块实现命令行操作
    args = parse_args()
    #3、递归取出dir_path下所有jpg文件,包括子文件夹的jpg文件
    all_pic_list = glob_all(args.dir_path)
    #4、for循环实现依次删除异常图片(包括损坏、非三通道)
    for i, img_path in enumerate(all_pic_list):
        try:

            ……

        except Exception:
            #检测到异常就删掉图片
            logging.warning('%s has broken. Delete it.' % img_path)
            #logging.error(traceback.format_exc())
            os.remove(img_path)
        

关注
打赏
1664196048
查看更多评论
立即登录/注册

微信扫码登录

0.0462s