(image1, image2, provided_image, output_folder, flow_output_folder, max_dimension, index)
| 86 | #AAAAAAAAAAAAAAAAAAA |
| 87 | |
| 88 | def process_image(image1, image2, provided_image, output_folder, flow_output_folder, max_dimension, index): |
| 89 | image1 = resize_image(image1, max_dimension) |
| 90 | image2 = resize_image(image2, max_dimension) |
| 91 | |
| 92 | flow = compute_optical_flow(cv2.cvtColor(image1, cv2.COLOR_BGR2GRAY), cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)) |
| 93 | warped_image = warp_image(provided_image, flow) |
| 94 | |
| 95 | warped_image_path = os.path.join(output_folder, f'warped_provided_image_{index + 1}.png') |
| 96 | save_image(warped_image, warped_image_path) |
| 97 | print(f"Warped image saved as '{warped_image_path}'") |
| 98 | |
| 99 | flow_image_path = os.path.join(flow_output_folder, f'optical_flow_{index + 1}.png') |
| 100 | save_optical_flow(flow, flow_image_path) |
| 101 | print(f"Optical flow map saved as '{flow_image_path}'") |
| 102 | |
| 103 | return warped_image |
| 104 | |
| 105 | def process_images(input_folder, output_folder, flow_output_folder, provided_image_path, max_dimension): |
| 106 | if not os.path.exists(output_folder): |
no test coverage detected