Updated Experiment selection

Author: JensBlack

DeepLabStream.py

@@ -430,9 +430,11 @@ def calculate_fps(self, current_analysis_time):
     ##################
     @staticmethod
     def set_up_experiment():
-        # add custom experiment here
-        experiment_dict = {1: ExampleExperiment}
-        experiment = experiment_dict[EXP_NUMBER]()
+        import importlib
+        from utils.configloader import EXP_NAME
+        mod = importlib.import_module('experiments.experiments')
+        experiment_class = getattr(mod, EXP_NAME)
+        experiment = experiment_class()
         return experiment
 
     def start_experiment(self):

settings.ini

@@ -7,11 +7,11 @@ MULTIPLE_DEVICES = False
 CAMERA_SOURCE = 2
 
 [DeepLabCut]
-DLC_PATH = D:\maDeepLabCut\deeplabcut
-MODEL = 13PointsFeb3-trainset95shuffle1_MobileNetV2
+DLC_PATH = PATH_TO_DLC
+MODEL = MODEL_NAME
 
 [Experiment]
-EXP_NUMBER = 1
+EXP_NAME = ExampleExperiment
 RECORD_EXP = True
 
 [Video]

utils/configloader.py

@@ -51,7 +51,7 @@ def get_script_path():
 
 
 # experiment
-EXP_NUMBER = dsc_config['Experiment'].getint('EXP_NUMBER')
+EXP_NAME = dsc_config['Experiment'].get('EXP_NAME')
 RECORD_EXP = dsc_config['Experiment'].getboolean('RECORD_EXP')
 
 START_TIME = time.time()