README.md

metadata

title: Webtoon Lineart Cropper
emoji: 🤗
colorFrom: yellow
colorTo: blue
sdk: gradio
sdk_version: 5.25.2
app_file: app.py
python_version: 3.12.10
pinned: false

Helix-Painting Data Tool

웹툰 PSD 파일을 이미지 파일(PNG)로 변환하고, 컷 추출 및 인물 검출 등을 통해 데이터를 추출하는 도구입니다.

Prerequisites

Prerequisites

Setup a project

Setup a project

Features

• PSD → PNG 변환
  웹툰 PSD 파일을 고해상도 PNG 이미지로 변환합니다.
• 컷 추출 및 필터링
  이미지 내 흰색 영역을 기준으로 컷 박스를 추출하고, 인물 검출(face detector)과 태깅(tagger) 기능을 통해 컷 별 데이터를 생성합니다.
• 병렬 처리
  처리 속도를 높이기 위해 Python의 multiprocessing을 사용한 병렬 이미지 처리 기능을 제공합니다.

Usage

1. PSD 파일을 PNG 이미지로 변환

convert_psd_to_png.py 스크립트를 사용하여 PSD 파일들을 PNG 이미지로 변환합니다.

python scripts/convert_psd_to_png.py --directory <PSD_directory> --output <output_directory> [--visible_layers layer1 layer2 ...] [--invisible_layers layer3 layer4 ...]

• --directory : PSD 파일을 검색할 디렉토리
• --output : 변환된 PNG 파일을 저장할 디렉토리
• --visible_layers / --invisible_layers : 보이거나 숨길 레이어들을 지정

2. 이미지 컷 추출 및 필터링

parse_cut_from_page.py 스크립트를 사용하여 컷 박스를 추출하고, 각 컷에 대해 필터링된 데이터를 생성합니다.

python scripts/parse_cut_from_page.py --lineart <lineart_directory> --flat <flat_directory> --segmentation <segmentation_directory> --color <color_directory> --output <output_directory> [--num_process <number_of_processes>]

• --lineart : 라인아트 이미지가 저장된 디렉토리
• --flat : 채색 전 평면 이미지가 저장된 디렉토리
• --segmentation : 세분화 이미지가 저장된 디렉토리
• --color : 컬러 이미지가 저장된 디렉토리
• --output : 잘라낸 컷 이미지를 저장할 디렉토리
• --num_process : 병렬 처리에 사용할 프로세스 수 (선택값)

3. 태깅 및 필터링 실행

run_tag_filter.py 스크립트를 사용하여 이미지에 태깅 및 필터링을 수행하고, 결과를 저장합니다.

python scripts/run_tag_filter.py --input_dir <input_directory> --output_dir <output_directory> [--ext png jpg jpeg]

• --input_dir : 필터링할 이미지들이 저장된 디렉토리
• --output_dir : 필터링 결과 이미지 및 캡션 파일을 저장할 디렉토리
• --ext : 처리할 이미지 확장자 목록 (기본값: png)

---

이와 같이 각 스크립트는 별도의 인자들을 받아서 작업을 수행하며, 스크립트 내부의 로직에 따라 PSD 변환, 컷 추출, 인물 검출/태깅, 캡션 파일 생성 등의 기능을 제공합니다.

4. Gradio 데모 페이지

아래 명렁어를 실행하여 Gradio 데모 페이지로 원하는 PNG파일을 업로드하여 필터링된 이미지 컷 추출 결과를 ZIP파일 혹은 PSD파일로 확인할 수 있습니다.

python app.py