计算两幅图片的farneback 稠密光流，并将结果图显示出来的程序

最新推荐文章于 2024-09-12 01:15:00 发布

vola9527

最新推荐文章于 2024-09-12 01:15:00 发布

阅读量6.1k

点赞数 2

分类专栏：算法&概念&工具文章标签： python optflow

本文链接：https://blog.csdn.net/vola9527/article/details/38259999

版权

算法&概念&工具专栏收录该内容

50 篇文章 0 订阅

订阅专栏

import cv2
import numpy as np
import Image
import cv2.cv as cv


def image_joint(image_list,opt):#opt= vertical ,horizon
    image_num=len(image_list)
    image_size=image_list[0].size
    height=image_size[1]
    width=image_size[0]
    
    if opt=='vertical':
        new_img=Image.new('RGB',(width,image_num*height),255)
    else:
        new_img=Image.new('RGB',(image_num*width,height),255)
    x=y=0
    count=0
    for img in image_list:
        
        new_img.paste(img,(x,y))
        count+=1
        if opt=='horizontal':
            x+=width
        else : y+=height
    return new_img






start_img = Image.open("C:/Users/Dell/Desktop/output/0260.bmp")
later_img = Image.open("C:/Users/Dell/Desktop/output/0275.bmp")
print type(start_img)
start_img=np.array(start_img)
later_img=np.array(later_img)
print start_img.shape
prvs = cv2.cvtColor(start_img,cv2.COLOR_BGR2GRAY)
next = cv2.cvtColor(later_img,cv2.COLOR_BGR2GRAY)
print next.shape




flow = cv2.calcOpticalFlowFarneback(prvs,next,0.5, 3, 15,3, 5, 1.2, 0)


hsv = np.zeros_like(start_img)
hsv[...,1] =255
mag, ang = cv2.cartToPolar(flow[...,0], flow[...,1])
hsv[...,0] = ang*180/np.pi/2
hsv[...,2] = cv2.normalize(mag,None,0,255,cv2.NORM_MINMAX)
RGB= cv2.cvtColor(hsv,cv2.COLOR_HSV2BGR)
GRAY = cv2.cvtColor(RGB,cv2.COLOR_BGR2GRAY)
print type(GRAY)
GRAY=Image.fromarray(GRAY)
start_img=Image.fromarray(start_img)
later_img=Image.fromarray(later_img)


joint_image=image_joint([start_img,later_img,GRAY],'vertical')
joint_image.show()
#GRAY.show()
#cv2.imshow('frame2',RGB)


cv.WaitKey()


#if ord('s'):
   
#    cv2.imwrite('opticalhsv.png',GRAY)