ytcollector 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

ytcollector/__init__.py

@@ -3,7 +3,7 @@ SBS Dataset Collector - YouTube 영상 수집 및 YOLO-World 검증 파이프라
 """
 from pathlib import Path
 
-__version__ = "1.0.1"
+__version__ = "1.0.3"
 __author__ = "SBS Dataset Team"
 
 # Package root directory

ytcollector/config.py

@@ -23,7 +23,7 @@ def get_paths(base_dir: Path = None):
 
 # 사용자 지정 출력 경로 (네트워크 드라이브 등)
 # macOS에서는 "/Volumes/Data/..." 등으로 마운트된 경로를 사용해야 함
-NAS_PATH_WINDOWS = r"\\192.168.100.192\Data\Private Dataset\SBS_De-Identification_YouTube"
+NAS_PATH_WINDOWS = r"\\NAS_SERVER_IP\Data\Private Dataset\SBS_De-Identification_YouTube"
 NAS_PATH_MAC = "/Volumes/Data/Private Dataset/SBS_De-Identification_YouTube"
 
 if platform.system() == 'Windows':

ytcollector/downloader.py

@@ -85,6 +85,15 @@ class VideoDownloader:
             'download_ranges': download_range_func(None, [(start_sec, end_sec)]),
             'force_keyframes_at_cuts': True,
         }
+
+        # Use ffmpeg from imageio-ffmpeg
+        try:
+            import imageio_ffmpeg
+            ffmpeg_path = imageio_ffmpeg.get_ffmpeg_exe()
+            ydl_opts['ffmpeg_location'] = ffmpeg_path
+            logger.debug(f"Using ffmpeg from: {ffmpeg_path}")
+        except ImportError:
+            logger.warning("imageio-ffmpeg not found, relying on system ffmpeg")
         
         logger.info(f"Downloading segment [{start_sec}s - {end_sec}s] from: {url}")
         

{ytcollector-1.0.1.dist-info → ytcollector-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ytcollector
-Version: 1.0.1
+Version: 1.0.3
 Summary: SBS 데이터셋 수집기
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
@@ -8,6 +8,7 @@ Requires-Dist: yt-dlp>=2024.1.0
 Requires-Dist: ultralytics>=8.1.0
 Requires-Dist: opencv-python>=4.9.0
 Requires-Dist: tqdm>=4.66.0
+Requires-Dist: imageio-ffmpeg>=0.4.9
 
 # Downloader: SBS 데이터셋 수집기
 
@@ -17,7 +18,7 @@ YouTube 영상에서 얼굴, 자동차 번호판, 타투, 텍스트 자막을
 
 **필수 요구사항:**
 - Python 3.8 이상
-- FFmpeg (Mac: `brew install ffmpeg`)
+- FFmpeg (Mac: `brew install ffmpeg`) **👈 필수 설치! (영상 자르기에 필요)**
 
 **설치:**
 ```bash
@@ -28,7 +29,7 @@ pip install -e .
 **프로젝트 초기화 (최초 1회):**
 필요한 폴더(`urls/`, `video/`)를 생성합니다.
 ```bash
-downloader init
+ytcollector init
 ```
 
 ---
@@ -54,13 +55,13 @@ face,https://www.youtube.com/watch?v=VIDEO_ID,2,30,설명
 ### 기본 다운로드 (순차 실행)
 안정적으로 하나씩 다운로드하고 검증합니다.
 ```bash
-downloader download --task face
+ytcollector download --task face
 ```
 
 ### 🚀 Fast 모드 (병렬 다운로드)
 대량의 영상을 빠르게 수집할 때 사용합니다. (4개 스레드 동시 실행)
 ```bash
-downloader download --task face --fast
+ytcollector download --task face --fast
 ```
 * **방화벽 우회**: 랜덤 딜레이(1~3초)가 적용되어 차단을 방지합니다.
 * **에러 무시**: 중간에 에러가 나도 멈추지 않고 다음 영상으로 넘어갑니다.
@@ -73,13 +74,13 @@ downloader download --task face --fast
 
 ## 4. NAS / 네트워크 저장 설정
 
-영상을 로컬이 아닌 NAS(`\\192.168.100.192...`)에 저장하려면 설정을 수정하세요.
+영상을 로컬이 아닌 NAS에 저장하려면 설정을 수정하세요.
 OS(Windows/Mac)를 자동 감지하여 적절한 경로를 사용합니다.
 
-**설정 파일 수정:** `downloader/config.py`
+**설정 파일 수정:** `ytcollector/config.py`
 ```python
 # Windows 예시
-NAS_PATH_WINDOWS = r"\\192.168.100.192\Data\Private Dataset\..."
+NAS_PATH_WINDOWS = r"\\NAS_SERVER_IP\Data\Private Dataset\..."
 
 # Mac 예시 (/Volumes로 마운트된 경로 확인 필요)
 NAS_PATH_MAC = "/Volumes/Data/Private Dataset/..."
@@ -91,11 +92,11 @@ NAS_PATH_MAC = "/Volumes/Data/Private Dataset/..."
 
 | 명령어 | 설명 | 예시 |
 |--------|------|------|
-| `init` | 프로젝트 초기화 | `downloader init` |
-| `download` | 텍스트 파일 목록 대량 다운로드 | `downloader download --task face --fast` |
-| `download-single` | URL 1개만 테스트 다운로드 | `downloader download-single --task face ...` |
-| `verify` | 수동 YOLO 검증 (기존 파일) | `downloader verify --task face` |
-| `list-tasks` | 지원하는 태스크 목록 확인 | `downloader list-tasks` |
+| `init` | 프로젝트 초기화 | `ytcollector init` |
+| `download` | 텍스트 파일 목록 대량 다운로드 | `ytcollector download --task face --fast` |
+| `download-single` | URL 1개만 테스트 다운로드 | `ytcollector download-single --task face ...` |
+| `verify` | 수동 YOLO 검증 (기존 파일) | `ytcollector verify --task face` |
+| `list-tasks` | 지원하는 태스크 목록 확인 | `ytcollector list-tasks` |
 
 ## 6. 문제 해결
 * **검증 실패가 너무 많음**: `config.py`에서 `CONFIDENCE_THRESHOLD` (기본 0.25) 값을 낮춰보세요.

ytcollector-1.0.3.dist-info/RECORD

@@ -0,0 +1,12 @@
+config/settings.py,sha256=RcK41kaUC0zam5SsdXfb7u_qjM_TlJDa0a8hC_MGacQ,1197
+ytcollector/__init__.py,sha256=_8OQMfsVmKuF_bE2yzu9G63znAORYbljmSH_hLRiXS4,365
+ytcollector/cli.py,sha256=lhWIraLdPIXS1c35jakPULGuSwzslNA3sOnY8pOWfuM,7570
+ytcollector/config.py,sha256=FrRXle52EP-0OExt4QFuwaPps7ewXe9FC9oobSd1ZWE,2393
+ytcollector/downloader.py,sha256=2pZFFU2mcdELNIH2VC14Ndoq6LYx8uL5ENhQ_kLf5TU,13556
+ytcollector/utils.py,sha256=gInDx6adV-SfQ2SH5_i8w1gvYL-Nsmz1e1W__gCdVH8,4654
+ytcollector/verifier.py,sha256=8Nn3b6fTQYxCGPt01kJMDSZ2hy8gk54deSayOpBuY48,6286
+ytcollector-1.0.3.dist-info/METADATA,sha256=8bOkHNTS2wBwpToauXWK5uqS8dPPWKYrjO4YFOQi0AU,3543
+ytcollector-1.0.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ytcollector-1.0.3.dist-info/entry_points.txt,sha256=PoanZbxogGnV4tLcZZkla0Yh7OvPtqcukDYr563w5RA,53
+ytcollector-1.0.3.dist-info/top_level.txt,sha256=TVfBZHJgYRfSSTgLJELvOoMA55qR8kWuxtiIaItwzIQ,19
+ytcollector-1.0.3.dist-info/RECORD,,

ytcollector-1.0.1.dist-info/RECORD

@@ -1,12 +0,0 @@
-config/settings.py,sha256=RcK41kaUC0zam5SsdXfb7u_qjM_TlJDa0a8hC_MGacQ,1197
-ytcollector/__init__.py,sha256=EKsxEpFheA3CPDMu5by4zxbd0DdLlus_t8Ac7quzxG4,365
-ytcollector/cli.py,sha256=lhWIraLdPIXS1c35jakPULGuSwzslNA3sOnY8pOWfuM,7570
-ytcollector/config.py,sha256=QAZ2USv3jrXA1xk1SOyTpbig7SIA9ZhI4ObI7qRX1pw,2395
-ytcollector/downloader.py,sha256=gFw8tYkpiTK3aJibto15jdWM3UsdQSLRKuIK5dlEFck,13184
-ytcollector/utils.py,sha256=gInDx6adV-SfQ2SH5_i8w1gvYL-Nsmz1e1W__gCdVH8,4654
-ytcollector/verifier.py,sha256=8Nn3b6fTQYxCGPt01kJMDSZ2hy8gk54deSayOpBuY48,6286
-ytcollector-1.0.1.dist-info/METADATA,sha256=nZI0a3Y0bonUmd4pazk2Ys3n7uJwGB4NtHD9bGrzPAU,3470
-ytcollector-1.0.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-ytcollector-1.0.1.dist-info/entry_points.txt,sha256=PoanZbxogGnV4tLcZZkla0Yh7OvPtqcukDYr563w5RA,53
-ytcollector-1.0.1.dist-info/top_level.txt,sha256=TVfBZHJgYRfSSTgLJELvOoMA55qR8kWuxtiIaItwzIQ,19
-ytcollector-1.0.1.dist-info/RECORD,,