kernel_er = np.ones((kernel[0], kernel[0]), np.uint8) kernel_dil = np.ones((kernel[1], kernel[1]), np.uint8) kernel_head = np.ones((9, 9), np.uint8) # ---------------------- # Select videos to save # ---------------------- file_loc = file_loc + save_vid_name fourcc = cv2.VideoWriter_fourcc( *'XVID') #LJPG for lossless, XVID or MJPG works for compressed # write videos as selected above if write_images: normal_video, normalized_video, cropped_mouse, stereo_input_L, stereo_input_R, threeD_combined, threeD_smooth = \ write_videos(file_loc, write_images, do_not_overwrite, fourcc, frame_rate, width, height, crop_size, border_size, write_normal_video, write_normalized_video, write_cropped_mice, write_stereo_inputs, write_3D_combined, write_3D_smooth) # save data for further analysis if video_type == '2D': save_2D_data = True if save_data == True: fourcc = cv2.VideoWriter_fourcc(*'LJPG') #LJPG for lossless data_file = file_loc + '_data.avi' if os.path.isfile(data_file) and do_not_overwrite: raise Exception('File already exists') if save_2D_data and video_type == '2D': data_video_2D = cv2.VideoWriter(data_file, fourcc, frame_rate, (crop_size, crop_size), False) elif save_2D_data and not video_type == '2D': data_video_3D = cv2.VideoWriter(data_file, fourcc, frame_rate, (crop_size, crop_size), False)
# ---------------------- # Select videos to save # ---------------------- session_name_tag_cur = session_name_tag + '_' + str(v) save_file_location_cur = save_folder_location + session_name_tag + "\\" + session_name_tag_cur if save_videos or save_data: print("saving to " + save_file_location_cur) fourcc_videos = cv2.VideoWriter_fourcc( *"XVID") # LJPG for lossless, XVID for compressed # save the videos selected to be saved above if save_videos: normal_video, normalized_video, cropped_mouse, stereo_input, stereo_input_R, threeD_combined, threeD_smooth = \ write_videos(save_file_location_cur, save_videos, do_not_overwrite, fourcc_videos, frame_rate_saved_video, width, height, crop_size, crop_size, write_normal_video, write_normalized_video, write_cropped_mouse_video, False, False, False) """------------------------------------------------------------------------------------------------------------------------------------- ------------------------ Perform analysis ------------------------------------------------------------ --------------------------------------------------------------------------------------------------------------------------------------""" while True: # grab the frame ret, frame = vid.read() frame_num = vid.get(cv2.CAP_PROP_POS_FRAMES) if frame_num >= end_frame - 1: print("fin") break if ret: # ---------------------------------------------------------