def main_flir_image_processor(): """ Process an image taken with a Flir One Pro thermal camera. The image can be taken either with the official app or a custom app using the Flir's SDK. This does not work on videos splitted frame by frame, because the camera doesn't save thermal information when recording a video. I have left a picture to test out the script. """ input_file = 'test_images/' + 'flir_20190617T163823.jpg' fie = FlirImageExtractor() fie.process_image(input_file, upsample_thermal=True, transform_rgb=True) # fie.plot() rgb_image = fie.get_rgb_np() rgb_image = cv2.cvtColor(rgb_image, cv2.COLOR_BGR2RGB) thermal_image_3d = fie.img_thermal_rgb thermal_image_3d = cv2.cvtColor(thermal_image_3d, cv2.COLOR_BGR2RGB) # Creating region contours image_processor = ImageProcessor(option='dlib_68landmarks') image_processor.process_image(rgb_image) cv2.imshow("RGB image with contours", rgb_image) # cv2.waitKey(0) image_processor.apply_saved_contours(thermal_image_3d) cv2.imshow("Thermal image with contours", thermal_image_3d) cv2.waitKey(0) # thermal_image_raw = fie.get_thermal_np() # regions = image_processor.all_regions # print('Region\tMean T\tStd T') # # for region in regions: # mean, std = region.get_mean_std_temperature(thermal_image_raw) # print('{}\t{}\t{}'.format(region.name, mean, std)) # fie.export_thermal_to_csv('thermals_csv/'+file_name+'_thermal_csv.csv') fie.save_images()
os.path.join(folder_path, file_name))) image_names = [img for img in file_names if img.endswith(file_type)] video_writters = [] # Prepare all video writers for fps in fps_options: video_writters.append( cv2.VideoWriter(video_name.format(fps), 0, fps, (output_dim, output_dim))) for image_name in tqdm(image_names): image = cv2.imread(os.path.join(folder_path, image_name)) try: image_processor.process_image(image) except NoFaceFoundException: continue for video_writter in video_writters: video_writter.write(image) for video_writter in video_writters: video_writter.release() cv2.destroyAllWindows() # Code for displaying a single image # image_name = "IMG_20190607_020523.jpg" # image_name = "IMG_20190715_162030.jpg" # image_name = "IMG_20190706_140033_1.jpg"