Как делать снимки с видео мышкой - PullRequest
Новая
       22

Как делать снимки с видео мышкой

0 голосов
/ 31 января 2012

У меня есть видео с автомобилями на шоссе, поэтому я хотел бы вырезать один за другим с помощью мыши.Но мне нужно сделать это более автоматически, например, один из способов изменить фрейм: нажать клавишу пробела, вырезать мышью и автоматически сохранить изображение в папке.

Вы когда-нибудь видели что-то подобное дляLinux?Я пытаюсь сделать в OpenCV, хотя это не обязательно.

Заранее спасибо.

1 Ответ

2 голосов
/ 31 января 2012

Я добавил код ниже.Он отлично работает для меня.Нарисуйте прямоугольник вокруг объекта, который вы хотите сохранить отдельно.Нажмите «Пробел», чтобы перейти к следующему кадру (иногда вам нужно нажимать дольше, чтобы двигаться вперед. Я не знаю почему).Аналогично 'Esc' для выхода.

использование: python car.py видеофайл 

# car.py
import cv,sys
print "usage: python car.py <videofile>"

box=[0,0,0,0]
i = 0
global image
#   creating mouse callback function
def my_mouse_callback(event,x,y,flags,param):
    global drawing_box,i
    if event==cv.CV_EVENT_LBUTTONDOWN:
        events = event
        #print event
        drawing_box=True
        [box[0],box[1],box[2],box[3]]=[x,y,0,0]
        print x,y
    if event==cv.CV_EVENT_LBUTTONUP:
        events = event
        drawing_box=False
        events= event
        if box[2]<0:
            box[0]+=box[2]
            box[2]*=-1
        if box[3]<0:
            box[1]+=box[3]
            box[3]*=-1
        rect = (box[0],box[1],box[2],box[3])
        #print rect
        roi = cv.GetSubRect(image,rect)
        filename = str(i)+'.jpg'
        #save(image,rect,filename)
        cv.SaveImage(filename,roi)
        i = i+1 

    if event==cv.CV_EVENT_MOUSEMOVE:
        if (drawing_box==True):
            box[2]=x-box[0]
            box[3]=y-box[1] 


#   function to draw the rectangle, added flag -1 to fill rectangle. If you don't want to fill, just delete it.     
def draw_box(img,box):
    cv.Rectangle(img,(box[0],box[1]),(box[0]+box[2],box[1]+box[3]),(255,0,0),2)

#   main program    
capture = cv.CaptureFromFile(sys.argv[1])
image2 = cv.QueryFrame(capture)
drawing_box=False

#   make a clone of image
temp=cv.CloneImage(image2)

cv.NamedWindow("Box Example")
cv.SetMouseCallback("Box Example",my_mouse_callback,temp)

while(1):
#   global image
    image = cv.QueryFrame(capture)
    while(1):
        roi = 0
        save = False
        events = 0
        boxsaver = 0
        cv.Copy(image,temp)

        if drawing_box==True:
            draw_box(temp,box)

            i = i+1

        cv.ShowImage("Box Example",temp)
        if cv.WaitKey(20)==27:
            sys.exit()
        elif cv.WaitKey(20)==32:
            break
...