# 2. Extract a projected boundary of the image bbox = boundary(restored_image, eo, R, ground_height, pixel_size, focal_length) print("--- %s seconds ---" % (time.time() - start_time)) gsd = (pixel_size * (eo[2] - ground_height)) / focal_length # unit: m/px # Boundary size boundary_cols = int((bbox[1, 0] - bbox[0, 0]) / gsd) boundary_rows = int((bbox[3, 0] - bbox[2, 0]) / gsd) print('projectedCoord') start_time = time.time() proj_coords = projectedCoord(bbox, boundary_rows, boundary_cols, gsd, eo, ground_height) print("--- %s seconds ---" % (time.time() - start_time)) # Image size image_size = np.reshape(restored_image.shape[0:2], (2, 1)) print('backProjection') start_time = time.time() backProj_coords = backProjection(proj_coords, R, focal_length, pixel_size, image_size) print("--- %s seconds ---" % (time.time() - start_time)) print('resample') start_time = time.time() b, g, r, a = resample(backProj_coords, boundary_rows,
def rectify(path, name): print(path, name) ground_height = 0 # unit: m height_threshold = 100 # unit: m focal_length = 5 / 1000 # unit: m pixel_size = 0.00375 / 1000 # unit: m/px # 190911 R_CB = np.array( [[0.990635238726878, 0.135295782209043, 0.0183541578119133], [-0.135993334134149, 0.989711806459606, 0.0444561944563446], [-0.0121505910810649, -0.0465359159242159, 0.998842716179817]], dtype=float) dst_path = path + "/" + name + '/odm_orthophoto/' if not (os.path.isdir(dst_path)): os.mkdir(dst_path) dst_path2 = path + "/" + name + '/odm_individual_orthophoto/' if not (os.path.isdir(dst_path2)): os.mkdir(dst_path2) file_list = [] for root, dirs, files in os.walk(path + "/" + name + '/images'): files.sort() count = 0 len_files = len(files) for file in files: filename = os.path.splitext(file)[0] extension = os.path.splitext(file)[1] file_path = root + '/' + file if extension == '.JPG' or extension == '.jpg': print('Read the image - ' + file) image = cv2.imread(file_path, -1) elif extension == '.txt': print('Read EOP - ' + file) eo = read_eo(file_path, 3) eo = convertCoordinateSystem(eo) if eo[2] - ground_height <= height_threshold: print("The height of the image is too low: ", eo[2] - ground_height, " m") return "The height of the image is too low" # System Calibration OPK = calibrate(eo[3], eo[4], eo[5], R_CB) eo[3:] = OPK print(eo) if abs(OPK[0]) > 30 * (np.pi / 180) or abs( OPK[1]) > 30 * (np.pi / 180): print('Too much omega/phi will kill you') return R = Rot3D(eo) # 4. Extract a projected boundary of the image bbox = boundary(image, eo, R, ground_height, pixel_size, focal_length) # 5. Compute GSD & Boundary size # GSD gsd = (pixel_size * (eo[2] - ground_height)) / focal_length # unit: m/px print("GSD: ", gsd) # Boundary size boundary_cols = int((bbox[1, 0] - bbox[0, 0]) / gsd) boundary_rows = int((bbox[3, 0] - bbox[2, 0]) / gsd) print(boundary_rows, boundary_cols) # 6. Compute coordinates of the projected boundary proj_coords = projectedCoord(bbox, boundary_rows, boundary_cols, gsd, eo, ground_height) # Image size image_size = np.reshape(image.shape[0:2], (2, 1)) # 6. Back-projection into camera coordinate system backProj_coords = backProjection(proj_coords, R, focal_length, pixel_size, image_size) # 7. Resample the pixels b, g, r, a = resample(backProj_coords, boundary_rows, boundary_cols, image) # 8. Create GeoTiff dst = dst_path2 + filename createGeoTiff(b, g, r, a, bbox, gsd, boundary_rows, boundary_cols, dst) # file_list.append("../../" + dst + '.tif') # in relative path file_list.append(dst + '.tif') # in absolute path count += 1 progressbc.send_update(count / len_files * 100 * 0.9) # 10. Mosaic individual orthophotos for each band working_path1 = './OTB-7.0.0-Linux64/' working_path2 = './bin/' set_env = './otbenv.profile' mosaic_execution = './otbcli_Mosaic' start_time = time.time() os.chdir(working_path1) # change path # https://stackoverflow.com/questions/13702425/source-command-not-found-in-sh-shell/13702876 subprocess.call(set_env, shell=True) os.chdir(working_path2) # subprocess.call("ls", shell=True) print(file_list) # subprocess.call(mosaic_execution + ' -il ' + ' '.join(file_list) + # ' -comp.feather ' + 'large' + # ' -out ' + dst_path + 'IMG_RGB.tif', shell=True) # subprocess.call(mosaic_execution + ' -il ' + ' '.join(file_list) + # ' -out ' + "../../" + dst_path + 'odm_orthophoto.tif', shell=True) # in relative path subprocess.call(mosaic_execution + ' -il ' + ' '.join(file_list) + ' -out ' + dst_path + 'odm_orthophoto.tif', shell=True) # in absolute path print("--- %s seconds ---" % (time.time() - start_time)) progressbc.send_update(100) print("Sent!")