Delete transparent_process.py

transparent_process.py

@@ -1,90 +0,0 @@
-'''
-python transparent_process.py Star_Rail_Tribbie_MMD_Videos_30s_Green Star_Rail_Tribbie_MMD_Videos_30s_Transparent --max_workers=10
-'''
-
-import os
-import sys
-import argparse
-from moviepy.editor import VideoFileClip, ImageSequenceClip
-from PIL import Image
-import numpy as np
-from concurrent.futures import ThreadPoolExecutor
-
-def green_to_transparent(frame):
-    """
-    将绿色背景转化为透明背景
-    """
-    try:
-        pil_image = Image.fromarray(frame)
-        # 将图像转换为RGBA模式
-        pil_image = pil_image.convert("RGBA")
-        # 获取图像的像素数据
-        data = pil_image.getdata()
-        # 创建一个新的像素列表
-        new_data = []
-        for item in data:
-            # 判断是否为绿色（这里使用简单的RGB范围判断）
-            if item[0] < 100 and item[1] > 200 and item[2] < 100:
-                # 将绿色像素设置为透明
-                new_data.append((item[0], item[1], item[2], 0))
-            else:
-                # 保留其他像素
-                new_data.append(item)
-        # 更新图像的像素数据
-        pil_image.putdata(new_data)
-        return np.array(pil_image)
-    except Exception as e:
-        print(f"Error processing frame: {e}")
-        return frame
-
-def process_video(video_path, output_path, max_workers=10):
-    """
-    处理视频，将绿色背景转化为透明背景（多线程版本）
-    """
-    try:
-        video = VideoFileClip(video_path)
-        fps = video.fps
-        audio = video.audio
-        frames = list(video.iter_frames(fps=fps))
-        
-        processed_frames = []
-        
-        # 使用线程池处理帧
-        with ThreadPoolExecutor(max_workers=max_workers) as executor:
-            futures = [executor.submit(green_to_transparent, frame) for frame in frames]
-            for future in futures:
-                processed_frame = future.result()
-                processed_frames.append(processed_frame)
-        
-        # 将处理后的帧保存为视频
-        processed_video = ImageSequenceClip(processed_frames, fps=fps)
-        processed_video = processed_video.with_audio(audio)
-        processed_video.write_videofile(output_path, codec="png")  # 使用PNG编码支持透明度
-        print(f"Video saved to {output_path}")
-    except Exception as e:
-        print(f"Error processing video {video_path}: {e}")
-
-def main(input_folder, output_folder, max_workers=10):
-    """
-    遍历输入文件夹中的所有视频，处理并保存到输出文件夹（多线程版本）
-    """
-    if not os.path.exists(output_folder):
-        os.makedirs(output_folder)
-    
-    for video_file in os.listdir(input_folder):
-        if video_file.endswith((".mp4", ".avi", ".mov")):
-            video_path = os.path.join(input_folder, video_file)
-            output_path = os.path.join(output_folder, video_file)
-            print(f"Processing {video_path}...")
-            process_video(video_path, output_path, max_workers)
-            print(f"Finished processing {video_path}")
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="Convert green background to transparent in videos.")
-    parser.add_argument("input_folder", type=str, help="Path to the folder containing input videos.")
-    parser.add_argument("output_folder", type=str, help="Path to the folder where processed videos will be saved.")
-    parser.add_argument("--max_workers", type=int, default=10, help="Number of workers for parallel processing.")
-    
-    args = parser.parse_args()
-    
-    main(args.input_folder, args.output_folder, args.max_workers)