ytcollector 1.2.0__tar.gz → 1.2.3__tar.gz

ytcollector-1.2.3/MANIFEST.in

@@ -0,0 +1,2 @@
+include ytcollector/models/*.pt
+recursive-include ytcollector *.pt

{ytcollector-1.2.0 → ytcollector-1.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ytcollector
-Version: 1.2.0
+Version: 1.2.3
 Summary: YouTube 콘텐츠 수집기 - 얼굴, 번호판, 타투, 텍스트 감지
 Author: YTCollector Team
 License: MIT

{ytcollector-1.2.0 → ytcollector-1.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ytcollector"
-version = "1.2.0"
+version = "1.2.3"
 description = "YouTube 콘텐츠 수집기 - 얼굴, 번호판, 타투, 텍스트 감지"
 readme = "README.md"
 requires-python = ">=3.8"
@@ -59,6 +59,9 @@ Repository = "https://github.com/yourusername/ytcollector"
 where = ["."]
 include = ["ytcollector*"]
 
+[tool.setuptools.package-data]
+ytcollector = ["models/*.pt"]
+
 [tool.black]
 line-length = 100
 target-version = ['py38']

{ytcollector-1.2.0 → ytcollector-1.2.3}/ytcollector/__init__.py

@@ -17,12 +17,12 @@ CLI 사용 예시:
     ytc -c face text --fast
 """
 
-from .config import CATEGORY_NAMES, CATEGORY_QUERIES, USER_AGENTS, LICENSE_PLATE_PATTERNS
+from .config import CATEGORY_NAMES, CATEGORY_QUERIES, USER_AGENTS, LICENSE_PLATE_PATTERNS, LICENSE_PLATE_MODEL_PATH
 from .analyzer import VideoAnalyzer, check_dependencies
 from .downloader import YouTubeDownloader
 from .cli import run, main as cli_main
 
-__version__ = "1.1.8"
+__version__ = "1.2.3"
 __all__ = [
     # 주요 클래스
     "VideoAnalyzer",
@@ -32,6 +32,7 @@ __all__ = [
     "CATEGORY_QUERIES",
     "USER_AGENTS",
     "LICENSE_PLATE_PATTERNS",
+    "LICENSE_PLATE_MODEL_PATH",
     # 유틸리티
     "check_dependencies",
     "run",

{ytcollector-1.2.0 → ytcollector-1.2.3}/ytcollector/analyzer.py

@@ -32,7 +32,7 @@ try:
 except ImportError:
     USE_GPU = False
 
-from .config import LICENSE_PLATE_PATTERNS, YOLO_MODEL_NAME, YOLO_CONFIDENCE
+from .config import LICENSE_PLATE_PATTERNS, LICENSE_PLATE_MODEL_PATH, YOLO_CONFIDENCE
 
 
 class VideoAnalyzer:
@@ -65,7 +65,7 @@ class VideoAnalyzer:
                 if self.yolo_model is None:
                     device = "cuda" if USE_GPU else "cpu"
                     print(f"  YOLO 모델 로딩 중... (Device: {device})")
-                    self.yolo_model = YOLO(YOLO_MODEL_NAME)
+                    self.yolo_model = YOLO(str(LICENSE_PLATE_MODEL_PATH))
                     self.yolo_model.to(device)
 
     def extract_frames(self, video_path, num_frames=10):

{ytcollector-1.2.0 → ytcollector-1.2.3}/ytcollector/config.py

@@ -1,4 +1,9 @@
 # 설정 상수
+from pathlib import Path
+
+# 모델 경로 (패키지 내부)
+MODEL_DIR = Path(__file__).parent / "models"
+LICENSE_PLATE_MODEL_PATH = MODEL_DIR / "license_plate_detector.pt"
 
 USER_AGENTS = [
     'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
@@ -57,8 +62,18 @@ CATEGORY_NAMES = {
 # 카테고리별 제외 키워드 (제목에 포함 시 스킵)
 BLACKLIST_KEYWORDS = {
     'tattoo': [
-        "눈썹", "입술", "두피", "영구",  # 반영구 시술 제외
-        "립타투", "헤어타투", "SMP"
+        # 반영구 화장 (Semi-permanent makeup)
+        "눈썹", "입술", "두피", "헤어라인", "아이라인", "구렛나룻",
+        "반영구", "영구화장", "립타투", "헤어타투", "SMP",
+        "마이크로블레이딩", "microblading", "엠보", "콤보", "PMU",
+        # 임시/가짜 타투
+        "헤나", "henna", "스티커", "페이크", "fake", "임시", "붙이는",
+        # 타투 제거
+        "제거", "레이저", "지우기", "removal",
+        # 도안/디자인만 (실제 시술 아님)
+        "도안", "디자인", "스케치", "드로잉", "그리기",
+        # 의료/미용 시술
+        "유두", "유륜", "흉터", "점",
     ],
     'face': [],
     'license_plate': [],
@@ -66,8 +81,7 @@ BLACKLIST_KEYWORDS = {
 }
 
 # YOLO 설정
-YOLO_MODEL_NAME = 'license_plate_detector.pt'  # 번호판 전용 학습 모델
-YOLO_CONFIDENCE = 0.5                          # 오탐지 방지를 위해 신뢰도 상향
+YOLO_CONFIDENCE = 0.5  # 오탐지 방지를 위해 신뢰도 상향
 
 # 번호판 정규식 패턴 (한국 자동차 번호판 중심)
 LICENSE_PLATE_PATTERNS = [

{ytcollector-1.2.0 → ytcollector-1.2.3}/ytcollector.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ytcollector
-Version: 1.2.0
+Version: 1.2.3
 Summary: YouTube 콘텐츠 수집기 - 얼굴, 번호판, 타투, 텍스트 감지
 Author: YTCollector Team
 License: MIT

{ytcollector-1.2.0 → ytcollector-1.2.3}/ytcollector.egg-info/SOURCES.txt

@@ -1,3 +1,4 @@
+MANIFEST.in
 README.md
 pyproject.toml
 ytcollector/__init__.py
@@ -12,4 +13,5 @@ ytcollector.egg-info/SOURCES.txt
 ytcollector.egg-info/dependency_links.txt
 ytcollector.egg-info/entry_points.txt
 ytcollector.egg-info/requires.txt
-ytcollector.egg-info/top_level.txt
+ytcollector.egg-info/top_level.txt
+ytcollector/models/license_plate_detector.pt