Example #1
0
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    
    




Example #2
0
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)