import requests import json from PIL import Image import os def getImgCoordinatesByMagi(image_path:"包含后缀的文件路径") -> "全部对话坐标list,失败返回none": # headers = { # 'authority': 'rogerxavier-fastapi-t5-magi.hf.space', # 'method': 'GET', # 'scheme': 'https', # 'Accept': '*/*', # 'Accept-Encoding': 'gzip, deflate, br, zstd', # 'Accept-Language': 'zh-CN,zh;q=0.9', # 'Cookie': 'spaces-jwt=eyJhbGciOiJFZERTQSJ9.eyJyZWFkIjp0cnVlLCJwZXJtaXNzaW9ucyI6eyJyZXBvLmNvbnRlbnQucmVhZCI6dHJ1ZX0sIm9uQmVoYWxmT2YiOnsia2luZCI6InVzZXIiLCJfaWQiOiI2NDJhNTNiNTE2ZDRkODI5M2M5YjdiNzgiLCJ1c2VyIjoicm9nZXJ4YXZpZXIifSwiaWF0IjoxNzE2NjExMTE3LCJzdWIiOiIvc3BhY2VzL3JvZ2VyeGF2aWVyL2Zhc3RhcGlfdDVfbWFnaSIsImV4cCI6MTcxNjY5NzUxNywiaXNzIjoiaHR0cHM6Ly9odWdnaW5nZmFjZS5jbyJ9.W00jo8kiRgwCpq5aaGhaPE2RP2jLOyvfimjyIfHVhP1gs7NHkBkRzVTFqYv3TRtZoHNPTiFiI5Ehu12KP06sDQ', # 'Dnt': '1', # 'Priority': 'u=1, i', # 'Sec-Ch-Ua': '"Chromium";v="124", "Google Chrome";v="124", "Not-A.Brand";v="99"', # 'Sec-Ch-Ua-Mobile': '?0', # 'Sec-Ch-Ua-Platform': '"Windows"', # 'Sec-Fetch-Dest': 'empty', # 'Sec-Fetch-Mode': 'cors', # 'Sec-Fetch-Site': 'same-origin', # 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36' # } with open(image_path, 'rb') as file: image_bytes = file.read() files = { "image": image_bytes, } magi_fastapi_base = 'https://rogerxavier-fastapi-t5-magi.hf.space/getCoordinates' try: # resp = requests.post(magi_fastapi_base, files=files, headers=headers) ##是仓库public后好像就api可用了 resp = requests.post(magi_fastapi_base, files=files) ##不使用header看看会不会还限制 print(resp.json()) return resp.json()["panels"] except Exception as e: print(e) return None #意外情况返回 def path_to_format(old_path)->"比如/path/to/your/image/1.jpg ->/path/to/your/image/100.jpg 经过03d": # 获取文件名和文件扩展名 file_path, file_name = os.path.split(old_path) file_name, file_extension = os.path.splitext(file_name) # 提取文件名中的数字部分(假设数字部分在文件名的末尾) number_str = ''.join(filter(str.isdigit, file_name)) number = int(number_str) # 格式化数字部分为03d格式 formatted_number = "{:03d}".format(number) # 生成新的文件名 new_file_name = f"{file_name.replace(number_str, formatted_number)}{file_extension}" new_absolute_path = os.path.join(file_path, new_file_name) return new_absolute_path def save_img(new_save_path:"新文件的保存路径(包含后缀)",old_img_path:"旧文件路径(包含后缀)")->"void生成新的文件保存 ,传入旧文件路径是为了删除有问题的旧文件": print(new_save_path) # 原始照片文件名 original_image = old_img_path #打开原始照片 # img = Image.open(original_image) # text_bboxes_for_all_images = getImgCoordinatesByMagi(original_image) # if text_bboxes_for_all_images is not None: # for index, box in enumerate(text_bboxes_for_all_images): # cropped_img = img.crop(tuple(box)) # # 保存裁剪后的图片,并按照数字大小依次排序命名(包括本身的也要3d,比如 0.jpg - >000.000_cropped.jpg) # # cropped_img.save(f"{os.path.splitext(new_save_path)[0]}_{index:03d}_cropped.jpg") # else: # print("图片识别有问题,准备删除") # os.remove(original_image) ##防止文件打开无法删除 text_bboxes_for_all_images = getImgCoordinatesByMagi(original_image) if text_bboxes_for_all_images is not None: with Image.open(original_image) as img: for index, box in enumerate(text_bboxes_for_all_images): cropped_img = img.crop(tuple(box)) # 保存裁剪后的图片,并按照数字大小依次排序命名(包括本身的也要3d,比如 0.jpg - >000.000_cropped.jpg) cropped_img.save(f"{os.path.splitext(new_save_path)[0]}_{index:03d}_cropped.jpg") else: print("图片识别有问题,准备删除") os.remove(original_image) if __name__ == '__main__': # 获取需要裁剪的无水印漫画位置 img_path = 'manga1' subdir_path = os.path.join(os.getcwd(), img_path) # 图片素材获取(包含子目录下所有图片) image_files = [] for root, dirs, files in os.walk(subdir_path): for file in files: if file.endswith(".jpg") or file.endswith(".png"): image_files.append(os.path.relpath(os.path.join(root, file))) # 创建处理后的子目录在与image_files同级目录下 processed_subdir_path = os.path.join(os.path.dirname(subdir_path), f"{img_path}2")#加2表示经过2阶段处理 os.makedirs(processed_subdir_path, exist_ok=True) # 对image_files进行某种处理,生成新图片,并保存在处理后的子目录中 for img_file in image_files: # 处理图片的代码(这里仅作示例) # 假设处理后的图片为new_img img_dir = os.path.dirname(img_file) new_img_dir = os.path.join(processed_subdir_path, img_dir) os.makedirs(new_img_dir, exist_ok=True) new_img_path = os.path.join(new_img_dir, os.path.basename(img_file)) new_img_path = path_to_format(new_img_path) if not os.path.exists(f"{os.path.splitext(new_img_path)[0]}_{1:03d}_cropped.jpg"): # 如果已经处理过那么跳过 ->只看第一个数字片段就行 # 处理图片并保存 save_img(new_save_path=new_img_path, old_img_path=img_file) else: print(f"Skipping {new_img_path} as it already exists.")