유튜브에서 원하는 객체를 검출하여 이미지로 저장해주는 코드입니다.
- OS
ubuntu 20.04 에서 테스트를 진행했습니다. - GPU
RTX3090 에서 테스트를 진행했습니다. - Tensorflow
tensorflow 2.0 이상 에서 작동합니다.
- clone repogitory
git clone https://github.com/kyeul611/gity.git
cd ./gity
- 모듈 설치
pip install -r requirements.txt
- chromedrive
apt-get update
apt install chromium-chromedriver
- youtube-dl
wget https://yt-dl.org/downloads/latest/youtube-dl -O youtube-dl
chmod a+rx youtube-dl
-
detection-weight
이곳에서 yolov3.tf.data-00000-of-00001 파일을 다운받아 gity/yolov3_tf2 폴더 안으로 옮겨주세요.
터미널에서 아래 명령어를 실행
python main.py --keyword [키워드] --class_name [클래스 종류]
Detection Class
Object-detector는 COCO Dataset의 에 대해서 작동합니다.
탐지 가능한 클래스 종류는 이곳에서 확인 가능합니다.
example:
python main.py --keyword tusker\ elephant --class_name elephant --num_video 5
추가 인자는 아래 명령어를 통해 확인하세요.
python main.py --help
다음 Open source를 참고하였습니다.
{
"박 결" : "[email protected]",
"정민혁" : "[email protected]",
"이제헌" : "[email protected]",
}