[robots@robots](armor-lines) Fix import error

robots-at-robots/research/robots_at_robots/dataset/armor_lines_detector.py

@@ -7,9 +7,7 @@ from polystar.common.models.image import Image
 
 from polystar.common.models.image_annotation import ImageAnnotation
 from research.common.dataset.twitch.twitch_roco_datasets import TwitchROCODataset
-#from ..armor_dataset_factory import
-#research.robots_at_robots.dataset.armor_dataset_factory import ArmorDatasetFactory ?? Will not find ArmorDatasetFactory
-#from ..code/robots-at-robots/research/robots_at_robots/dataset/armor_dataset_factory.py import ArmorDatasetFactory
+from research.robots_at_robots.dataset.armor_dataset_factory import ArmorDatasetFactory
 
 
 class ArmorLinesDetector:
@@ -43,15 +41,24 @@ class ArmorLinesDetector:
 
 
 
+
 if __name__ == "__main__":
 
     lineDetector = ArmorLinesDetector()
+    armor_lines = []
+    for i, (armor_img, armor, k, p) in enumerate(ArmorDatasetFactory.from_dataset(TwitchROCODataset.TWITCH_470158483)):
+        armor_lines[i] = lineDetector.detect_lines(armor_img)
+   # test = cv2.imread('test.png')  # ouverture
+   # test = cv2.cvtColor(test, cv2.COLOR_BGR2RGB)  # conversion des couleurs
+   # armor_lines = lineDetector.detect_lines(test)
+   # plt.imshow(armor_lines[7], cmap='gray')
+   # plt.title('50 lignes principales')
+   # plt.show()
+
+    for armor in enumerate(armor_lines):
+        plt.imshow(armor)
+        plt.show()
+        plt.clf()
 
-    #for i, (armor_img, armor, k, p) in enumerate(ArmorDatasetFactory.from_dataset(TwitchROCODataset.TWITCH_470158483)):
-       # armor_lines = lineDetector.detect_lines(armor_img)
-    test = cv2.imread('test.png')  # ouverture
-    test = cv2.cvtColor(test, cv2.COLOR_BGR2RGB)  # conversion des couleurs
-    armor_lines = lineDetector.detect_lines(test)
-    plt.imshow(armor_lines, cmap='gray')
-    plt.title('50 lignes principales')
-    plt.show()
\ No newline at end of file
+        if i == 5:
+            break
\ No newline at end of file