Heatmap classification locations
Create heatmaps representing the locations of the classified behavior
[2]:
from simba.plotting.heat_mapper_clf_mp import HeatMapperClfMultiprocess
[10]:
#DEFINITIONS
CONFIG_PATH = '/Users/simon/Desktop/envs/troubleshooting/two_black_animals_14bp/project_folder/project_config.ini'
FRAME_SETTING = False
VIDEO_SETTING = True
FINAL_IMAGE_SETTING = True
BODY_PART = 'Nose_1'
CLASSIFIER_NAME = 'Attack'
CORES = 5
STYLE_ATTRIBUTES = {'palette': 'jet',
'shading': 'gouraud',
'bin_size': 100,
'max_scale': 'auto'}
DATA_PATHS = ['/Users/simon/Desktop/envs/troubleshooting/two_black_animals_14bp/project_folder/csv/machine_results/Together_1.csv']
[11]:
plotter = HeatMapperClfMultiprocess(config_path=CONFIG_PATH,
style_attr = STYLE_ATTRIBUTES,
final_img_setting=FINAL_IMAGE_SETTING,
video_setting=VIDEO_SETTING,
frame_setting=FRAME_SETTING,
bodypart=BODY_PART,
clf_name=CLASSIFIER_NAME,
core_cnt=CORES,
files_found=DATA_PATHS)
plotter.run()
Processing 1 video(s)...
SIMBA COMPLETE: Final classifier heatmap image saved at at /Users/simon/Desktop/envs/troubleshooting/two_black_animals_14bp/project_folder/frames/output/heatmaps_classifier_locations/Together_1_final_frm.png complete
Creating heatmaps, multiprocessing (chunksize: 1, cores: 5)...
Image 348/1738, Video 1/1...
Image 696/1738, Video 1/1...
Image 1044/1738, Video 1/1...
Image 1392/1738, Video 1/1...
Image 1740/1738, Video 1/1...
Joining Together_1 multiprocessed video...
SIMBA COMPLETE: Video concatenated (elapsed time: 11.2593s) complete
Heatmap video Together_1 complete (elapsed time: 101.2183s) ...
SIMBA COMPLETE: heatmap visualizations for {} videos created in project_folder/frames/output/heatmap_classifier locations directory (elapsed time: 101.2478s) complete
[ ]:
[ ]: