Study/class note
딥러닝 / object detection 다른 사진으로 학습시키는 방법
chanzae
2022. 5. 2. 01:21
ㅇobject detection을 다른 사진으로 직접 학습 시켜서 직접 구현하는 방법
1. object detection하기를 원하는 이미지를 2개 선별
(예: 눈 감은 사람과 눈 뜬 사람, 개가 하품하는 사진과 코를 핥는 사진, 사과와 파인애플, 뉴발란스와 나이키)
2. 웹스크롤링을 통해 데이터 수집
3. 라벨링 작업
c:\\object2밑에 파인애플 사진 100장, 사과 사진 100장을 넣어두세요.
그 다음에 아나콘다 프롬프트 창을 열어 아래의 명령어를 수행하시오.
그리고 주피터에서 파이썬 창을 연 뒤 아래의 코드를 실행하시오.
이미지명을 숫자 1번부터 200번까지 변경합니다.
import os
count = 1
for i in os.listdir():
if '.jpg' in i:
os.rename(i, str(count)+'.'+i.split('.')[-1])
count+=1
이미지를 라벨링을 하기 위해서 다음 github 에서 압축파일을 받고 압축을 풉니다.
https://github.com/tzutalin/labelImg
아나콘다 프롬프트창을 열고 압축을 푼 폴더로 이동합니다.
(snowdeer_env1) C:\Users\ejcej\Downloads\labelImg-master>labelImg
라벨 바꾸는 프로그램 실행 됨
labelImag 프로그램을 열고 View 에 Auto save mode 를 누룹니다. 그리고 아래에 2번 처럼 yolo 로 변경합니다.
4. 신경망 학습
5. 사진 object detection
6. 동영상 object detection
7. 실시간 웹캠 object detection
반응형