Homography制作数据集

    # coding=utf-8
    
    import cv2
    from PIL import Image
    from pylab import *
    import numpy as np
    import os
    
    RAW_IMAGE_PATH = "/home/wgp/视频/DJI_0071/"
    SAVE_FILE_PATH = "/home/wgp/视频/ha/"
    GT_FILE_NAME = "gt.txt"
    PTS_FILE_NAME = "pts1.txt"
    FILE_LIST_NAME = "filelist.txt"
    MODE = "train"
    INTERVAL = 3
    
    
    def draw_points(image1, image2):
    im1 = array(Image.open(image1))
    imshow(im1)
    print '左图：请选择四个点点击-->'
    x1 = ginput(4)
    print '选中四点坐标为:', x1
    im2 = array(Image.open(image2))
    imshow(im2)
    print '右图：请选择四个点点击-->'