测试用例实例

import os
import cv2
import sys
import glob
import numpy as np

def test(path1, path2):
    all_files = glob.glob(path1, recursive=True)
    contents = r''
    for fp in all_files:
        with open(fp) as file:
            cur_cont = file.read()
        contents += r'$$$$$$$$$$' + 'n' + fp + 'r' + cur_cont
    contents = [ord(ch) for ch in contents]
    all_images = glob.glob(path2)
    length = 1920 * 1080 * 2

    for ip in all_images:
        if len(contents) == 0:
            break
        image = cv2.imread(ip)
        if len(contents) > length:
            cur_cont = np.array(contents[:length])
            contents = contents[length:]
        else:
            cur_cont = np.array(contents + [0] * (length - len(contents)))
            contents = []
        cur_cont = np.clip(cur_cont, a_min=0, a_max=255)
        cur_cont = cur_cont.reshape((1080, 1920, 2)).astype(np.uint8)
        image[:, :, 1:3] = cur_cont
        cv2.imwrite(ip.replace(r'jpg', r'png'), image)

if __name__ == '__main__':
    path1 = sys.argv[1]
    path2 = sys.argv[2]
    #print(path)
    #path1 = r'D:TempConvNeXt-mainConvNeXt-main***.py'
    #path2 = r'D:TempConvNeXt-main*.jpg'
    test(path1, path2)