yolo v3 как извлечь изображение из обнаруженного объекта - PullRequest
Новая
       7

yolo v3 как извлечь изображение из обнаруженного объекта

0 голосов
/ 04 февраля 2019

Я работаю над проектом по обнаружению объектов, и я работаю над извлечением изображений объектов, которые обнаруживаются yolo v3 с помощью Anaconda.Вот как я установил Python и запустил yolo v3: https://github.com/reigngt09/yolov3workflow/tree/master/2_YoloV3_Execute Проблема в том, что у меня нет знаний по Python.Можно ли извлечь изображения и сохранить их в отдельном файле во время работы видео?

1 Ответ

0 голосов
/ 11 февраля 2019

Это один из доступных методов.

  1. Извлечение изображений из видео
  2. Функция обнаружения вызовов для каждого изображения
  3. Сохранение каждого изображения с ограничительной рамкой

под кодом, который я использовал, 

import cv2
import detect as dt
from darknet import Darknet
from PIL import Image

vidcap = cv2.VideoCapture('your/video/path.mp4')
success, image = vidcap.read()
count = 0

m = Darknet('your/cfg/file/path.cfg')
m.load_weights('weight/file/path.weights')
use_cuda = 1
m.cuda()

while success:
    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
    im_pil = Image.fromarray(image)
    im_pil = im_pil.resize((m.width, m.height))
    boxes = dt.do_detect(m, im_pil, 0.5, 0.4, use_cuda)

    result = open('your/save/file/path/frame%04d.txt'%(count), 'w')
    for i in range(len(boxes)):
        result.write(boxes[i])
    count = count + 1
    success, image = vidcap.read()
    result.close()
...