使用python随机取Reside数据集hazy文件夹中的图像并重命名另存于TrainA文件夹中
2022/2/23 17:22:05
本文主要是介绍使用python随机取Reside数据集hazy文件夹中的图像并重命名另存于TrainA文件夹中,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
今天想要对模型做去雾辅助训练,花了九牛二虎之力才下载好Reside的ITS训练集,它包含clear和hazy文件夹,其中clear包含1399张无雾的ground true。
而hazy包含13990张合成雾化图,也就是利用一张无雾图合成了10张不同雾化程度的图。
而我的需求是在10张不同雾化的图随机选一张出来与gt匹配做训练就行了。就需要写个小python来实现,无奈代码能力实在是太菜了,花了小半个下午才完成这个功能,还东查查西查查的才勉强用了一个巨笨的方法实现。。。不过不管了,能用就行!
写这个小代码也令我学习到一些函数的使用,比如shutil.copy(),和几个编辑文件路径名字的os函数。同时也敲响警钟,代码能力实在是太菜了,不能再做copy侠了。
以后也打算开始写写博客记录一下自己遇到的问题和解决的办法啦!顺便希望今天刚投出去PR能够顺利录用!
下面是代码
import os import glob import random import shutil # 图像重命名并另存于别的路径 def copy_image(image_file, save_path): file_name = os.path.basename(image_file) #重命的图像名字要与gt的名字一致,取最前面的数字就可以 save_name = file_name.split('_')[0] #重命名并另存为的命令 shutil.copy(image_file, save_path + '/' + save_name + '.png') return 0 #想要存的路径 save_path = 'TrainA' os.makedirs(save_path, exist_ok=True) # 取所有雾化图像 files = [] files += sorted(glob.glob(os.path.join('hazy/hazy') + "/*.*")) # 每张图的10张雾化图像只随机取1张,构建随机的索引利用列表索引去得到随机的图片 for i in range(len(files) // 10): a = i*10 b = random.randrange(10) c = a+b copy_image(files[c], save_path)
这篇关于使用python随机取Reside数据集hazy文件夹中的图像并重命名另存于TrainA文件夹中的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-05-08有遇到过吗?同样的规则 Excel 中 比Python 结果大
- 2024-03-30开始python成长之路
- 2024-03-29python optparse
- 2024-03-29python map 函数
- 2024-03-20invalid format specifier python
- 2024-03-18pool.map python
- 2024-03-18threads in python
- 2024-03-14python Ai 应用开发基础训练,字符串,字典,文件
- 2024-03-13id3 algorithm python
- 2024-03-13sum array elements python