pmd-python/tools/denoise_imgs.py

import cv2
import os
import glob
import numpy as np

img_folder = '/Users/junexie/workspace/lightek/pmdrecons-python/imgs/test_pj/20240829140109545'
save_folder = '/Users/junexie/workspace/lightek/pmdrecons-python/imgs/test_pj/picture_denoise'

# 如果保存文件夹不存在，则创建
if not os.path.exists(save_folder):
    os.makedirs(save_folder)
    print(f"创建文件夹: {save_folder}")


subfolders = os.listdir(img_folder)
count = 0
for img_path in os.listdir(os.path.join(img_folder, subfolders[0])):
    if img_path.endswith('.txt'): 
        continue
    imgs = []
    for sub in subfolders:
        print(os.path.join(img_folder, sub, img_path))
        count += 1
        img = cv2.imread(os.path.join(img_folder, sub, img_path))
        imgs.append(img)
        print(img.shape)
    
    assert len(imgs) == 10, len(imgs)
    print(count)
    imgs = np.array(imgs)
    img = np.mean(imgs, axis=0)

    cv2.imwrite(os.path.join(save_folder, img_path), img)