python学习

2021-01-28

• 以numpy数组格式读取图片和txt文件

import numpy as np
import cv2 as cv

root = "000050"

img_path = "./train/images/" + root + ".jpg"
txt_path = "./train/labels/" + root + ".txt"

src=cv.imread(img_path)       

loc = np.loadtxt(txt_path)  

cv.namedWindow('input_image', cv.WINDOW_AUTOSIZE)
cv.imshow('input_image', src)
cv.waitKey(0)
cv.destroyAllWindows()

• 写txt文件

f = open('./data/data_classes.txt', 'w')
j = 1
for i in a:
    f.write('    \"'+i+'\": '+str(j)+',\n')
    j += 1
f.close()

•  图像缩放，读取文件夹下所有文件名

from PIL import Image

path = '***'

fileList = os.listdir(path)

for img in fileList:
    
    #读取图像
    root = path + "/" + img
    im = Image.open(root)
    #im.show()
     
    #原图像缩放为128x128
    im_resized = im.resize((512, 384))
    #im_resized.show()
    
    root = "./resize_qdu_dataset/val/images/" + img
    im_resized.save(root)