1、数据集简述:
博主依据视频抽帧得到的图像数据集为基础,进行相应的人工数据清洗;因为博主观察到图像中有一些是不符合项目要求的图像,为了方便接下来的人工标注,必须剔除掉不符合要求的图像。
视频抽帧得到的图像:
![微信截图_20210609104141](https://i-blog.csdnimg.cn/blog_migrate/33eed5f422dd4e59d15c598ea193bf43.jpeg)
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窗口,每次操作后都会弹出,直到全部图片筛选完成:
![微信截图_20210608225430](https://i-blog.csdnimg.cn/blog_migrate/456e29a7d374ac827193fc04353d9332.png)
图片符合要求并且按键操作正确:
![微信截图_20210608225430](https://i-blog.csdnimg.cn/blog_migrate/2074c9899bc1a95d8a2b9e43de08e648.png)
图片不符合要求并且按键操作正确:
![微信截图_20210608225430](https://i-blog.csdnimg.cn/blog_migrate/4f8de06cc22ac5896b7f9cdb4f8e7fea.png)
按键操作失误:
![微信截图_20210608225430](https://i-blog.csdnimg.cn/blog_migrate/7cdcce212de3c6a5ac1b95e95ee56ce1.png)