ytcollector 1.1.9__py3-none-any.whl → 1.2.3__py3-none-any.whl

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/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/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,14 +62,18 @@ CATEGORY_NAMES = {
 # 카테고리별 제외 키워드 (제목에 포함 시 스킵)
 BLACKLIST_KEYWORDS = {
     'tattoo': [
-        "두피 문신", "두피문신",
-        "눈썹 문신", "눈썹문신",
-        "입술 문신", "입술문신",
-        "립타투", "립 타투",
-        "헤어타투", "헤어 타투",
-        "구레나룻문신", "구레나룻 문신",
-        "틴트 입술",
-        "반영구", "SMP"
+        # 반영구 화장 (Semi-permanent makeup)
+        "눈썹", "입술", "두피", "헤어라인", "아이라인", "구렛나룻",
+        "반영구", "영구화장", "립타투", "헤어타투", "SMP",
+        "마이크로블레이딩", "microblading", "엠보", "콤보", "PMU",
+        # 임시/가짜 타투
+        "헤나", "henna", "스티커", "페이크", "fake", "임시", "붙이는",
+        # 타투 제거
+        "제거", "레이저", "지우기", "removal",
+        # 도안/디자인만 (실제 시술 아님)
+        "도안", "디자인", "스케치", "드로잉", "그리기",
+        # 의료/미용 시술
+        "유두", "유륜", "흉터", "점",
     ],
     'face': [],
     'license_plate': [],
@@ -72,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.1.9.dist-info → ytcollector-1.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ytcollector
-Version: 1.1.9
+Version: 1.2.3
 Summary: YouTube 콘텐츠 수집기 - 얼굴, 번호판, 타투, 텍스트 감지
 Author: YTCollector Team
 License: MIT
@@ -44,7 +44,7 @@ Requires-Dist: ytcollector[analysis,dev]; extra == "all"
 pip install yt-dlp
 ```
 
-### 분석 기능용 패키지 (권장 - v1.1.9+)
+### 분석 기능용 패키지 (권장 - v1.2.0+)
 
 분석 기능을 원활하게 사용하려면 아래 패키지들이 필요합니다. GPU(CUDA)가 설치된 경우 자동으로 가속이 활성화됩니다.
 
@@ -192,9 +192,10 @@ https://www.youtube.com/watch?v=aqz-KE-bpKQ, 00:10, sample_task
 | 번호판 | YOLOv8 전용 모델 + OCR | v1.1.9: 번호판 전용 학습 모델 + 한국 번호판 패턴 매칭 |
 | 타투 | OpenCV HSV 분석 | 피부 영역 내 잉크 패턴 감지 |
 
-### 주요 최적화 (v1.1.5~1.1.9)
-- **번호판 전용 모델** (v1.1.9): YOLO-World → 번호판 전용 학습 모델로 교체 (감지율 대폭 향상, 모델 크기 27MB→6MB)
-- **오탐지 방지** (v1.1.9): YOLO 감지 + OCR 패턴 매칭 둘 다 만족해야 번호판 판정, ROI 크기 필터링 추가
+### 주요 최적화 (v1.1.5~1.2.0)
+- **자막 영역 필터링** (v1.2.0): 화면 하단 60%~95% 영역만 자막으로 인식, 상단 로고/워터마크 제외
+- **번호판 전용 모델** (v1.1.9): YOLO-World → 번호판 전용 학습 모델로 교체 (감지율 향상, 모델 크기 27MB→6MB)
+- **오탐지 방지**: YOLO 감지 + OCR 패턴 매칭 둘 다 만족해야 번호판 판정, ROI 크기 필터링 추가
 - **ROI 기반 감지**: 전체 화면이 아닌 YOLO가 지정한 영역만 OCR하여 속도와 정확도 향상
 - **GPU 가속 지원**: CUDA 사용 가능 시 YOLO 및 OCR 자동 가속
 - **로그 기반 중복 방지**: `youtube_url_*.txt` 기록을 참조하여 중복 분석 방지

ytcollector-1.2.3.dist-info/RECORD

@@ -0,0 +1,13 @@
+ytcollector/__init__.py,sha256=P4AIOCG2glB95wwN9QhUuF0HsgVxy6JyjTvkZKuQQXg,1152
+ytcollector/analyzer.py,sha256=zgtcw6pSCzJwa4NJ04XFJlLaoRt8CwXGxwuL9o8eduM,14414
+ytcollector/cli.py,sha256=aHF4EuQRPLKh65lnkI_dZ0ResztlVjpHlS5iHfzmpig,5577
+ytcollector/config.py,sha256=lpzYaODbOErUni_oAJ_zU0dpIQzy8gJDz-Cisrqf1tE,3646
+ytcollector/dataset_builder.py,sha256=nfArEwszoCln48n3T0Eff_4OOaYv8FF0YH8cARBGMWQ,2608
+ytcollector/downloader.py,sha256=TeC6agUmSPHZSZ9jdoc42i8i_NobzTEkoRtAIgW80kI,14544
+ytcollector/utils.py,sha256=6XDif-e3GbMHmUvTsBT0YblxNxYnS-2I8HnmjMBZs-M,4254
+ytcollector/models/license_plate_detector.pt,sha256=jsOyVKbIdhDwN6kJV0Ysr6EanAMiTjOijGodGsKsUbA,6241454
+ytcollector-1.2.3.dist-info/METADATA,sha256=u3Xz0naQgl9F-ZFOM7vw4fu25KxR28KfOPuz6eTzGkI,7319
+ytcollector-1.2.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ytcollector-1.2.3.dist-info/entry_points.txt,sha256=waiVuSJJYt-6_DAal-T4JkHgejo7wKYLdKrEI7tZ-ms,127
+ytcollector-1.2.3.dist-info/top_level.txt,sha256=wozNyCUm0eMOm-9U81yTql6oGaM2O5rWVBXDb93zzyQ,12
+ytcollector-1.2.3.dist-info/RECORD,,

ytcollector-1.1.9.dist-info/RECORD

@@ -1,12 +0,0 @@
-ytcollector/__init__.py,sha256=EaxBP_0Fv0LEFdg067uZxBrQHwOKGX8u08Y4b5uF1-Q,1094
-ytcollector/analyzer.py,sha256=7VJt4chc25HsEz8OwBDZhTz_8LnkpgSBM6mJKQpIUls,14391
-ytcollector/cli.py,sha256=aHF4EuQRPLKh65lnkI_dZ0ResztlVjpHlS5iHfzmpig,5577
-ytcollector/config.py,sha256=HmcFqMV1Z3kClnGKYi0q9cRZIXMRPAxIEI8cggatfrU,3199
-ytcollector/dataset_builder.py,sha256=nfArEwszoCln48n3T0Eff_4OOaYv8FF0YH8cARBGMWQ,2608
-ytcollector/downloader.py,sha256=TeC6agUmSPHZSZ9jdoc42i8i_NobzTEkoRtAIgW80kI,14544
-ytcollector/utils.py,sha256=6XDif-e3GbMHmUvTsBT0YblxNxYnS-2I8HnmjMBZs-M,4254
-ytcollector-1.1.9.dist-info/METADATA,sha256=WQIup61B02yB9ddtOaFGtlDYoV4nJVRD-vvVI_sVjc0,7208
-ytcollector-1.1.9.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-ytcollector-1.1.9.dist-info/entry_points.txt,sha256=waiVuSJJYt-6_DAal-T4JkHgejo7wKYLdKrEI7tZ-ms,127
-ytcollector-1.1.9.dist-info/top_level.txt,sha256=wozNyCUm0eMOm-9U81yTql6oGaM2O5rWVBXDb93zzyQ,12
-ytcollector-1.1.9.dist-info/RECORD,,