From 63750f070738591da92b4c12b52b0768582660a2 Mon Sep 17 00:00:00 2001
From: Yasmine Moumou <yasmine.moumou@polymtl.ca>
Date: Wed, 14 Oct 2020 17:25:05 -0400
Subject: [PATCH] [robots@robots](armor-lines) Fix import error
---
.../dataset/armor_lines_detector.py | 29 ++++++++++++-------
1 file changed, 18 insertions(+), 11 deletions(-)
diff --git a/robots-at-robots/research/robots_at_robots/dataset/armor_lines_detector.py b/robots-at-robots/research/robots_at_robots/dataset/armor_lines_detector.py
index 7541e8d..ccc62cd 100644
--- a/robots-at-robots/research/robots_at_robots/dataset/armor_lines_detector.py
+++ b/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
--
GitLab