导入相关库
import cv2 as cv
import matplotlib.pyplot as plt
import numpy as np
import random
import os
读取图片
def read_img(mpath,img_main):
"""读取图片"""
img_name=os.listdir(mpath)
imgs=[]
for name in img_name:
if name==img_main:
main=cv.imread(mpath+name)
block=1
else:
imgs.append(cv.imread(mpath+name))
return imgs,main,img_name
处理图片使其长宽一致
def new_imgs(imgs):
"""处理图片使其长宽一致"""
w=[]
h=[]
for img in imgs:
w.append(img.shape[0])