From d9d70b22a6d6c05b6ac4f1d577dac3ff744e0ce0 Mon Sep 17 00:00:00 2001
From: Yasmine Moumou <yasmine.moumou@polymtl.ca>
Date: Tue, 11 Aug 2020 21:13:48 -0400
Subject: [PATCH] [robots](armor-lines) Use enumerate
---
.../robots_at_robots/dataset/armor_lines_detector.py | 8 +++-----
1 file changed, 3 insertions(+), 5 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 bf50395..d9c9597 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,7 +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 ArmorDatasetFactory
+from armor_dataset_factory import ArmorDatasetFactory
#from ..code/robots-at-robots/research/robots_at_robots/dataset/armor_dataset_factory.py import ArmorDatasetFactory
@@ -40,8 +40,6 @@ class ArmorLinesDetector:
if __name__ == "__main__":
lineDetector = ArmorLinesDetector()
- armorFactory = ArmorDatasetFactory()
- armorData = armorFactory.from_dataset(TwitchROCODataset.TWITCH_470158483)
- for element in armorData:
- lineDetector.detect_lines(element[0]) # get first image to test
\ No newline at end of file
+ for i, (armor_img, armor, k, p) in enumerate(ArmorDatasetFactory.from_dataset(TwitchROCODataset.TWITCH_470158483)):
+ lineDetector.detect_lines(armor_img)
\ No newline at end of file
--
GitLab