tiod0611 / gity

gather_image_through_youtube

GITY

Gather Image Through Youtube

유튜브에서 원하는 객체를 검출하여 이미지로 저장해주는 코드입니다.

목차

• 환경
• 설치
• 실행
• 작성자

Environment

• OS
  ubuntu 20.04 에서 테스트를 진행했습니다.
• GPU
  RTX3090 에서 테스트를 진행했습니다.
• Tensorflow
  tensorflow 2.0 이상 에서 작동합니다.

Installation

0. clone repogitory

git clone https://github.com/kyeul611/gity.git
cd ./gity

1. 모듈 설치

pip install -r requirements.txt

2. chromedrive

apt-get update
apt install chromium-chromedriver

3. youtube-dl

wget https://yt-dl.org/downloads/latest/youtube-dl -O youtube-dl
chmod a+rx youtube-dl

Detector

1. detection-weight

   이곳에서 yolov3.tf.data-00000-of-00001 파일을 다운받아 gity/yolov3_tf2 폴더 안으로 옮겨주세요.

run

터미널에서 아래 명령어를 실행

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

references

다음 Open source를 참고하였습니다.

• video-preprocessing

• object detector

author

{
	"박 결" : "[email protected]",
	"정민혁" : "[email protected]",
	"이제헌" : "[email protected]",
}