Flask + rembg (u2net/isnet/birefnet) + Pillow + OpenCV. Tab Tách nền (camera, so sánh 3 mức, polygon crop, cache) + Tab Ghép Frame (danh sách frame/watermark, tỉ lệ object, nền màu/trong suốt). UI responsive. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> |
||
|---|---|---|
| assets | ||
| cache | ||
| objects | ||
| output | ||
| static | ||
| .gitignore | ||
| README.md | ||
| requirements.txt | ||
| run.sh | ||
| server.py | ||
README.md
Product Image Studio — Tách nền & Ghép Frame
Công cụ web xử lý ảnh sản phẩm cho ecom: tách nền + ghép frame/watermark, chạy local, miễn phí. Stack: Flask + rembg (U²-Net/ISNet/BiRefNet ONNX) + Pillow + OpenCV.
Tính năng
Tab 1 — Tách nền
- Xóa nền AI (chọn model: u2net / isnet / birefnet), chống cháy sáng giữ chi tiết mép.
- Chụp ảnh trực tiếp từ camera, hoặc upload nhiều ảnh / kéo-thả.
- So sánh 3 mức chất lượng (low/medium/high) chạy song song.
- Vùng chọn đa điểm (polygon) ôm sát object, chỉnh tự do để cắt ngoài vùng.
- Review zoom + lật ảnh gốc; cache cutout để tinh chỉnh không xử lý lại.
- Lưu object (1:1 trong suốt) vào thư viện.
Tab 2 — Ghép Frame
- Danh sách nhiều frame & watermark, chọn là preview đổi ngay.
- Chỉnh tỉ lệ object, vị trí/độ mờ/cỡ watermark, viền.
- Nền: trong suốt hoặc bảng màu / màu tùy chọn.
- Lưu ra file (PNG nếu trong suốt, JPG nếu nền màu).
UI responsive cho desktop / tablet / mobile.
Chạy nhanh
./run.sh
Script tự tạo virtualenv, cài dependencies và khởi động server tại http://localhost:8001. Lần đầu sẽ tải model ONNX (~170MB), sau đó chạy offline.
Thủ công
python3 -m venv .venv && source .venv/bin/activate
pip install -r requirements.txt
python server.py
Ghi chú
- Apple Silicon: server ép
CPUExecutionProvider(CoreML biên dịch u2net bị treo). cache/,output/,objects/,assets/là dữ liệu runtime (đã gitignore).- Camera trực tiếp cần
localhosthoặc HTTPS; qua IP LAN sẽ tự dùng camera gốc của máy.