[2]Yolo v3 python

# Stupid python path shit.
# Instead just add darknet.py to somewhere in your python path
# OK actually that might not be a great idea, idk, work in progress
# Use at your own risk. or don't, i don't care

import sys, os
sys.path.append(os.path.join(os.getcwd(),'python/'))

import darknet as dn
import pdb

dn.set_gpu(0)
net = dn.load_net("cfg/yolo-thor.cfg", "/home/pjreddie/backup/yolo-thor_final.weights", 0)
meta = dn.load_meta("cfg/thor.data")
r = dn.detect(net, meta, "data/bedroom.jpg")
print r

# And then down here you could detect a lot more images like:
r = dn.detect(net, meta, "data/eagle.jpg")
print r
r = dn.detect(net, meta, "data/giraffe.jpg")
print r
r = dn.detect(net, meta, "data/horses.jpg")
print r
r = dn.detect(net, meta, "data/person.jpg")
print r

다음의 코드를 다음과 같이 수정하여 진행하였다.

수정코드

#gps = 전송된 위치정보
#slice = 분석을 위해 동영상을 프레임단위로 자른 이미지
#gps, slice 폴더의 위치 = /Users/username/gps , /Users/username/slice
#코드의 실행위치 = /Users/username/darknet/yolo.py
#yolo 분석을 통해 탐지시 /Users/username/detection_result.txt 에 저장

import sys, os
sys.path.append(os.path.join(os.getcwd(),'python/'))
sys.path.append(os.getcwd().replace('darknet', ''))
sys.path.append(os.getcwd().replace('darknet', 'gps/'))
sys.path.append(os.getcwd().replace('darknet', 'slice/'))
import darknet as dn
import pdb

#dn.set_gpu(0)

net = dn.load_net("cfg/yolov3.cfg".encode('utf-8'), "ww/yolo_final.weights".encode('utf-8'), 0)
meta = dn.load_meta("cfg/coco.data".encode('utf-8'))
 
 
path = '../slice/'
file_list = os.listdir(path)
if ".DS_Store" in file_list:
    file_list.remove(".DS_Store")
 
total_file_num = len(file_list)
current_file_num = int()
 
f = open('../detection_result.txt', 'a')
    
for file_name in file_list:
    filepath = path + file_name
    r = dn.detect(net, meta, filepath)
 
    current_file_num = current_file_num + 1
    print(str(current_file_num) + '///' + str(total_file_num))
    
    if r:
        print(r)
        print(file_name)
        
        with open('../detection_result.txt', 'a') as fileobject:
            fileobject.write(file_name)
            fileobject.write('\n')
 
print("END")

해당 코드의 실행위치는 /Users/username/darknet/yolo.py 의 파이썬 파일을 만들어 실행하였다.

다른 위치에서 실행하여도 되지만 경로를 바꿔줘야하는 불편함이있어 해당 위치에서 실행하는 것은 권장한다.

slice 폴더의 경우에는 동영상 분석하기 위해 이미지를 프레임 단위로 자른 이미지들이 들어있는 폴더이며

gps 폴더는 도로를 주행하면서 1초에 30번 기록된 위치 정보를 저장하는 폴더이다.

해당 코드를 이용하여 여러장의 이미지가 있는 디렉토리를 입력 데이터로 넣어서 여러장의 이미지를 순차적으로 탐지를 실시한다.

만약 탐지가 완료되면 저와 같은 경우는 /Users/sungjin/detection_result.txt 파일이 아래와 같이 저장된다.

만약 오류가 발생한다면 아래 블로그를 참고하여 오류의 부분을 확인 후 해결해 나아가면 도움이 될 것 같다.

https://dudgns7675.tistory.com/9

Darknet detector 여러가지 오류 처리하기

Darknet의 detector 예제를 다른 프로그램에 가져다 쓰고 싶어서 원본 폴더에서 필요한 파일들만 복사를 하니 여러가지 오류가 발생했습니다... 열심히 구글검색도 하고, 없는 건 직접 찾아보기도 하

dudgns7675.tistory.com