一、對數據集進行重命名—按順序編號
# -*- coding:utf8 -*-
#coding:UTF-8
import os
class BatchRename():
'''
批量重命名文件夾中的圖片文件
'''
def __init__(self):
self.path = 'C:\CBSD68'
def rename(self):
filelist = os.listdir(self.path)
total_num = len(filelist)
i = 1
for item in filelist:
if item.endswith('.png' ) :
src = os.path.join(os.path.abspath(self.path), item)
dst = os.path.join(os.path.abspath(self.path), str(i) + '.png')
try:
os.rename(src, dst)
print ('converting %s to %s ...' % (src, dst))
i = i + 1
except:
continue
print ('total %d to rename & converted %d jpgs' % (total_num, i))
if __name__ == '__main__':
demo = BatchRename()
demo.rename()
二、對數據集進行修改圖片大小
#coding:UTF-8
from PIL import Image
import os.path
import glob
def convertjpg(jpgfile,outdir,width=260,height=260):
img=Image.open(jpgfile)
try:
new_img=img.resize((width,height),Image.BILINEAR)
new_img.save(os.path.join(outdir,os.path.basename(jpgfile)))
except Exception as e:
print(e)
for jpgfile in glob.glob(r"C:\Users\86150\Desktop\datasets\ImageNet400\*.png"): #讀取文件
convertjpg(jpgfile,r"C:\Users\86150\Desktop\datasets\IMage280") #保存文件位置
三、將數據集彩色圖片轉換為灰度圖像
#coding:UTF-8
from skimage import io,transform,color
import numpy as np
def convert_gray(f,**args):#圖片處理與格式化的函數
rgb=io.imread(f) #讀取圖片
gray=color.rgb2gray(rgb) #將彩色圖片轉換為灰度圖片? ??
#dst=transform.resize(gray,(64,64))? #調整大小,圖像分辨率為64*64? ??
return gray
datapath = r'C:\Users\86150\Desktop\datasets\CImageNet400'
str=datapath+'/*.png' #識別.jpg的圖像
coll = io.ImageCollection(str,load_func=convert_gray)#批處理
for i in range(len(coll)):
io.imsave(r'C:/Users/86150/Desktop/datasets/ImageNet400/'+np.str(i)+'.png',coll[i]) #保存圖片在d:/daate/date/
?
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元
