下载的藻类图像数据集中,图像数据为.tif格式,要求更改为.jpg格式。
藻类图像数据集
代码
import os
import cv2
root = "D:\Desktop\paper\pytorch_learning\deep-learning-for-image-processing-
master\data_set\Aqualitas_100taxa" #图像数据集的绝对路径
for filename in os.listdir(root):#遍历当前目录下的所有文件夹
path = root + "\\" + filename #每个文件夹的绝对路径
tif_list = [x for x in os.listdir(path) if x.endswith(".tif")] # 找到当前路径下的所有.tif文件
for num, i in enumerate(tif_list):
i = path + "\\" + i
img = cv2.imread(i, -1) # 这里选择-1,不进行转化
cv2.imwrite(i.split('.')[0] + ".jpg", img)
print('总共:', len(tif_list), '张,剩余:', len(tif_list) - num - 1, '张')
os.remove(i) #删除tif格式文件
全部代码
#-*- codeing = utf-8 -*-
#@Time : 2021/10/16 16:08
#@File : Tif.py
#@Software:PyCharm
import os
import cv2
# data_root = os.path.abspath(os.path.join(os.getcwd(), "../"))
# path = os.getcwd() #获取当前代码文件的路径
# path = os.path.join(root,"Thalassiosira_pseudonana")
# print(path)
root = "D:\Desktop\paper\pytorch_learning\deep-learning-for-image-processing-master\data_set\data\photos"
# path = "D:\\Desktop\\paper\\A-ls\\data\\Achnanthidium_pyrenaicum"
# root = "D:\Desktop\paper\pytorch_learning\deep-learning-for-image-processing-master\data_set\data\photos"
for filename in os.listdir(root):
print(filename)
path = root + "\\" + filename
print(path)
tif_list = [x for x in os.listdir(path) if x.endswith(".tif")] # 找到当前路径下的所有.tif文件
print(tif_list)
print(type(tif_list))
for num, i in enumerate(tif_list):
i = path + "\\" + i
img = cv2.imread(i, -1) # 这里选择-1,不进行转化
cv2.imwrite(i.split('.')[0] + ".jpg", img)
print('总共:', len(tif_list), '张,剩余:', len(tif_list) - num - 1, '张')
os.remove(i)
# 输出所有类别名称
# tif_list = [x for x in os.listdir(root)]
# print(tif_list)
#删除.db文件
# for filename in os.listdir(root):
# print(filename)
# path = root + "\\" + filename
#
# for i in os.listdir(path):
# if i.endswith(".db"):
# i = path + "\\" + i
# os.remove(i)