cameraMatrix = cameraenvironment.getCamMatrixCV2() #Matrix distortP = cameraenvironment.getDistortCoeffsCV2() #Distort #Write out camera calibration info to .txt file target1 = '../Examples/camenv_data/calib/KR3_2014_1.txt' FileHandler.writeCalibFile(matrix, tancorr, radcorr, target1) #Write homography data to .csv file FileHandler.writeHomogFile(hg, imn, destination + 'homography.csv') #Write results to file FileHandler.writeAreaFile(uvareas, xyzareas, imn, destination + 'areas.csv') FileHandler.writeAreaCoords(uvpts, xyzpts, imn, destination + 'uvcoords.txt', destination + 'xyzcoords.txt') #Create shapefiles FileHandler.writeAreaSHP(xyzpts, imn, destination + 'shpfiles/', proj) #Write all image extents and dems target4 = destination + 'outputimgs/' for i in range(len(areas)): plotAreaPX(uvpts[i], imgset[i].getImageCorr(cameraMatrix, distortP), show=True, save=target4 + 'uv_' + str(imn[i])) plotAreaXYZ(xyzpts[i], dem, show=True, save=target4 + 'xyz_' + str(imn[i])) #------------------------------------------------------------------------------ print('\n\nFINISHED')
#Write homography to file FileHandler.writeHomogFile(hg, imn, destination + 'homography.csv') #Create shapefiles target1 = destination + 'shpfiles/' proj = 32633 FileHandler.writeAreaSHP(xyzpts, imn, target1, proj) #Plot areas in image plane and as XYZ polygons target2 = destination + 'outputimgs/' for i in range(len(xyzpts)): plotAreaPX(uvpts[i], imgset[i].getImageCorr(cameraMatrix, distortP), show=True, save=None) plotAreaXYZ(xyzpts[i], dem, show=True, save=None) #---Alternative method for plotting image extents using original RGB images---- #Get original images in directory ims = sorted(glob.glob(camimgs)) #Get camera correction variables newMat, roi = cv2.getOptimalNewCameraMatrix(cameraMatrix, distortP, (5184, 3456), 1, (5184, 3456)) #Get corresponding xy pixel areas and images count = 1 for p, i in zip(uvpts, ims): x = [] y = []