Parcourir la source

Update of display simulation curves script

Jérôme BUISINE il y a 4 ans
Parent
commit
9440fdcedb
2 fichiers modifiés avec 10 ajouts et 4 suppressions
  1. 1 1
      README.md
  2. 9 3
      display/display_simulation_curves.py

+ 1 - 1
README.md

@@ -149,4 +149,4 @@ Once you have simulation done. Checkout your **threshold_map/%MODEL_NAME%/simula
 
 ## License
 
-[The MIT license](https://github.com/prise-3d/Thesis-NoiseDetection-26-attributes/blob/master/LICENSE)
+[The MIT license](LICENSE)

+ 9 - 3
display/display_simulation_curves.py

@@ -1,13 +1,19 @@
+# main imports
 import numpy as np
 import pandas as pd
 
-import matplotlib.pyplot as plt
 import os, sys, argparse
 
-from modules.utils.data import get_image_features
+# image processing imports
+import matplotlib.pyplot as plt
+
+# modules imports
+sys.path.insert(0, '') # trick to enable import of main folder module
 
-from modules.utils import config as cfg
+import custom_config as cfg
+from data_attributes import get_image_features
 
+# other variables
 learned_zones_folder = cfg.learned_zones_folder
 models_name          = cfg.models_names_list
 label_freq           = 6