Added a script to crop an image to fit the object size

This commit is contained in:
Egor Tsyganchuk
2024-12-02 19:12:43 +03:00
parent 8813ebc56a
commit a02f275cb3
2 changed files with 48 additions and 0 deletions

39
crop_to_object.py Normal file
View File

@@ -0,0 +1,39 @@
import os
import argparse
from PIL import Image
def crop_to_object(input_folder, output_folder=None):
# Если выходная папка не указана, используем входную
if output_folder is None:
output_folder = input_folder
# Проверяем, существует ли выходная папка, если нет, создаем ее
if not os.path.exists(output_folder):
os.makedirs(output_folder)
# Проходим по всем файлам в входной папке
for filename in os.listdir(input_folder):
if filename.endswith('.png'):
img_path = os.path.join(input_folder, filename)
img = Image.open(img_path)
# Получаем границы непрозрачных пикселей
bbox = img.getbbox()
if bbox:
# Обрезаем изображение
img_cropped = img.crop(bbox)
# Сохраняем обработанное изображение
img_cropped.save(os.path.join(output_folder, filename))
print(f"Processed {filename}")
def main():
parser = argparse.ArgumentParser(description="Crop PNG images to the size of the visible object.")
parser.add_argument('input_folder', type=str, help='Path to the input folder containing PNG images.')
parser.add_argument('output_folder', type=str, nargs='?', default=None, help='Path to the output folder to save processed images. If not provided, the input folder will be used.')
args = parser.parse_args()
crop_to_object(args.input_folder, args.output_folder)
if __name__ == "__main__":
main()

View File

@@ -40,3 +40,12 @@
- Путь к каталогу, в котором будут размещены фрагменты `.png`
- Ширина
- Длина
## Обрезка изображения по объекту
Скрипт `crop_to_object.py` предназначен для обрезки PNG изображений до размеров видимого объекта. Он удаляет прозрачные области изображения и сохраняет результат в указанной папке. Если выходная папка не указана, скрипт перезаписывает оригинальные файлы.
Скрипт принимает два аргумента командной строки:
- Папка входных изображений (обязательный аргумент) — путь к папке, содержащей PNG изображения, которые вы хотите обработать.
- Папка выходных изображений (необязательный аргумент) — путь к папке, в которую будут сохранены обработанные изображения. Если этот аргумент не указан, скрипт будет перезаписывать оригинальные файлы в папке входных изображений.