README.md

---
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](docs/PREREQUISITES.md)

## Setup a project

[Setup a project](docs/SETUP.md)

## Features

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

## Usage

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

```shell
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` 스크립트를 사용하여 컷 박스를 추출하고, 각 컷에 대해 필터링된 데이터를 생성합니다.

```shell
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` 스크립트를 사용하여 이미지에 태깅 및 필터링을 수행하고, 결과를 저장합니다.

```shell
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파일로 확인할 수 있습니다.

```shell
python app.py
```