小的工具代碼

1. 批量重命名

# coding:utf-8
import os
"""
批量修改文件名
"""

movie_name = os.listdir('A2B')

label = 1

for temp in movie_name:
    new_name = 'img_' + str(label) + '.jpg'
    label += 1
    print(new_name)

    os.rename('A2B/' + temp, 'A2B/' + new_name)

2. 批量修改文件尺寸

import os
import os.path
import cv2

def dirlist(path, allfile):
    filelist =  os.listdir(path) 
    for filename in filelist:  
        filepath = os.path.join(path, filename)  
        if os.path.isdir(filepath): 
            dirlist(filepath, allfile)  
        else:  
            if filename[-4:] == '.jpg':
                allfile.append(filepath)  
    return allfile

path='G:/day-night-128/train/A'
allfile = dirlist(path, [])
# print(allfile)

width = 128
height = 128

for filename in allfile:
    print(filename)
    im = cv2.imread(filename)
    im = cv2.resize(im,(width,height))
    cv2.imwrite(filename,im)

3. 圖像拼接

import cv2
import numpy as np
‘’‘
核心代碼
’‘’
#橫向拼接
img1 = cv2.imread("lover.jpg")
img2 = cv2.imread("lover.jpg")
img3 = cv2.imread("lover.jpg")
heng = np.hstack((img1, img2, img3))

#縱向拼接
img4 = cv2.imread("lover.jpg")
img5 = cv2.imread("lover.jpg")
img6 = cv2.imread("lover.jpg")
shu = np.vstack((img4, img5, img6))

cv2.imwrite("hstack.jpg", heng)
cv2.imwrite("vstack.jpg", shu)
cv2.imshow("hstack", heng)
cv2.imshow("vstack", shu)