用python+opencv实现数据清洗

1、数据集简述：

        博主依据视频抽帧得到的图像数据集为基础，进行相应的人工数据清洗；因为博主观察到图像中有一些是不符合项目要求的图像，为了方便接下来的人工标注，必须剔除掉不符合要求的图像。

视频抽帧项目博客地址

视频抽帧得到的图像：

2、代码介绍：

        博主的思路是利用cv2.imshow显示窗口的原理和cv2.waitKey读取键值的原理，进行每张图片的筛选；符合要求的按“o”，不符合要求的按“f”，除了这两个键其余的键均归类为误触操作；三种情况均会把图片保存至相应的文件夹。

筛选之后，若误触操作文件夹内存在图片，可直接筛选或类推代码原理进行二轮数据清洗。

import os
import cv2
from tqdm import tqdm

#总数据集图像路径
imgway_1="./video_image"
#符合条件的图像的路径
imgway_2="./final_image"
#不符合条件的图像的路径
imgway_3="./out_image"
#按错键的情况，将错过的图像保存至规定路径
imgway_4='./mistake'


for filename in tqdm(os.listdir(r"%s" % imgway_1)):

    img = cv2.imread("%s/%s"%(imgway_1,filename))
    img = cv2.resize(img,(800,600))
    cv2.imshow('datasets', img)

    k=cv2.waitKey(0)& 0xFF
    if k == ord('o'):
        cv2.imwrite('%s/%s' % (imgway_2, filename),img)
        print('%s符合要求！' % filename)
        cv2.destroyAllWindows()

    elif k == ord('f'):
        cv2.imwrite('%s/%s' % (imgway_3, filename),img)
        print('%s不符合要求！' % filename)
        cv2.destroyAllWindows()

    else:
        cv2.imwrite('%s/%s' % (imgway_4, filename),img)
        print('误触错误的键！%s保存至mistake文件夹！' % filename)
        cv2.destroyAllWindows()
        

3、代码效果：

代码运行后，会弹出opencv的imshow窗口，每次操作后都会弹出，直到全部图片筛选完成：

图片符合要求并且按键操作正确：

图片不符合要求并且按键操作正确：

按键操作失误：