python粘贴图

import os
import random
import cv2
import time
import numpy as np
from PIL import Image
obj_imgs_path = "G:/dataset/coco128/images/train2017"
src_imgs_path = "G:/dataset/gofishing3"


src_imgslist = os.listdir(src_imgs_path)
obj_imgslist = os.listdir(obj_imgs_path)

# choseimg = random.choice(obj_imgslist)
# print("choseimg==",choseimg)

for i in range(len(src_imgslist)):
    src_img_path = src_imgs_path + "/"+ src_imgslist[i]
    srcimg= cv2.imread(src_img_path)

    choseimg = random.choice(obj_imgslist) # 随机从目标文件夹中抽取一张图片
    obj_img_path = obj_imgs_path + "/"+ choseimg
    objimg = cv2.imread(obj_img_path)
    # print("srcimg======",src_img_path,srcimg)
    try:
        srcimg_width = srcimg.shape[1]
        srcimg_height = srcimg.shape[0]
    except Exception as e:
        cv2.imwrite("./out/" + src_imgslist[i], pasteimg)
        continue

    try:
        objimg_width = objimg.shape[1]
        objimg_height = objimg.shape[0]
    except Exception as e:
        continue

    scale_factor = 2
    resized_image = objimg
    while (srcimg_width > objimg_width) or (srcimg_height > objimg_height):
        print("srcimg_width srcimg_height======", srcimg_width,srcimg_height)
        print("objimg_width objimg_height======", objimg_width, objimg_height)

        resized_image = cv2.resize(objimg, (0, 0), fx=scale_factor, fy=scale_factor)
        objimg_width = objimg_width * scale_factor
        objimg_height = objimg_height * scale_factor
        scale_factor = scale_factor + 1

    resized_image[:srcimg_height, :srcimg_width, :] = srcimg

    pasteimg = resized_image

    cv2.imwrite("./out/"+src_imgslist[i], pasteimg)