Fix sort bad patterns

sashacmc · sashacmc · commit 3df706620a69 · 2020-02-28T01:56:20.000+01:00
diff --git a/encqualitycont.py b/encqualitycont.py
@@ -168,6 +168,8 @@ def sort_patterns(self, files, out_folder):
         for path, pred in zip(images, self.test(images)):
             splitted = path.split(os.path.sep)
             name = splitted[-2]
+            if name == 'bad':
+                name = splitted[-3]
 
             res = pred[0]
             if res < 0.5:
