rat_summary_table_path = [r'F:/Videogame_Assay/AK_33.2_Pt.csv', 'F:/Videogame_Assay/AK_40.2_Pt.csv', 'F:/Videogame_Assay/AK_41.1_Pt.csv','F:/Videogame_Assay/AK_41.2_Pt.csv', 'F:/Videogame_Assay/AK_46.1_behaviour_only.csv', 'F:/Videogame_Assay/AK_48.1_IrO2.csv', 'F:/Videogame_Assay/AK_48.3_behaviour_only.csv', 'F:/Videogame_Assay/AK_48.4_IrO2.csv', 'F:/Videogame_Assay/AK_49.1_behaviour_only.csv','F:/Videogame_Assay/AK_49.2_behaviour_only.csv', 'F:/Videogame_Assay/AK_50.1_behaviour_only.csv', 'F:/Videogame_Assay/AK_50.2_behaviour_only.csv'] #colours = ['#FF0000','#FF8C00','#FF69B4','#BA55D3','#4B0082','#0000FF','#00BFFF','#2E8B57','#32CD32', '#ADFF2F','#7FFFD4','#FFDAB9'] RAT_ID = ['AK 33.2', 'AK 40.2', 'AK 41.1', 'AK 41.2', 'AK 46.1', 'AK 48.1','AK 48.3','AK 48.4', 'AK 49.1', 'AK 49.2','AK 50.1','AK 50.2'] for count, rat in enumerate(rat_summary_table_path): try: Level_1 = prs.Level_1_paths(rat) sessions_subset = Level_1 start_end_idx(sessions_subset) print(rat) print(count) except Exception: print (rat + '/error') continue
import os #os.sys.path.append('/home/kampff/Repos/Pac-Rat/libraries') os.sys.path.append('D:/Repos/Pac-Rat/libraries') import glob import numpy as np import matplotlib.pyplot as plt import cv2 import tracking_library as tracking import parser_library as prs # Reload modules hardrive_path = r'F:/' rat_summary_table_path = 'F:/Videogame_Assay/AK_48.4_IrO2.csv' Level_0 = prs.Level_0_paths(rat_summary_table_path) Level_1 = prs.Level_1_paths(rat_summary_table_path) Level_2_pre = prs.Level_2_pre_paths(rat_summary_table_path) Level_3 = prs.Level_3_pre_paths(rat_summary_table_path) Level_2_post = prs.Level_2_pre_paths(rat_summary_table_path) # Reload modules import importlib importlib.reload(tracking) session_list = Level_2_post for session in session_list: video_path = os.path.join(hardrive_path, session + '/Video.avi') try: # Open video video = cv2.VideoCapture(video_path)