Skip to content
Snippets Groups Projects
Commit e60729aa authored by Mathieu Beligon's avatar Mathieu Beligon
Browse files

[robots] (refactor) Move research into a robots_at_robots folder

parent fc2303c2
No related branches found
No related tags found
No related merge requests found
Showing
with 18 additions and 13 deletions
from typing import List
from research.dataset.armor_color_dataset_factory import ArmorColorDatasetGenerator
from research.robots_at_robots.dataset import ArmorColorDatasetGenerator
from research_common.dataset.directory_roco_dataset import DirectoryROCODataset
from research_common.image_pipeline_evaluation.image_pipeline_evaluation_reporter import ImagePipelineEvaluationReporter
from research_common.image_pipeline_evaluation.image_pipeline_evaluator import ImagePipelineEvaluator
......
......@@ -4,7 +4,8 @@ from polystar.common.image_pipeline.classifier_image_pipeline import ClassifierI
from polystar.common.image_pipeline.image_featurizer.mean_rgb_channels_featurizer import MeanChannelsFeaturizer
from polystar.common.image_pipeline.models.random_model import RandomModel
from polystar.common.image_pipeline.models.red_blue_channels_comparison_model import RedBlueComparisonModel
from research.armor_color.armor_color_pipeline_reporter_factory import ArmorColorPipelineReporterFactory
from research.robots_at_robots.armor_color.armor_color_pipeline_reporter_factory import \
ArmorColorPipelineReporterFactory
from research_common.dataset.twitch.twitch_roco_datasets import TwitchROCODataset
if __name__ == "__main__":
......
from typing import Iterable, List
from research.dataset.armor_digit_dataset_factory import ArmorDigitDatasetGenerator
from research.robots_at_robots.dataset import ArmorDigitDatasetGenerator
from research_common.dataset.directory_roco_dataset import DirectoryROCODataset
from research_common.image_pipeline_evaluation.image_pipeline_evaluation_reporter import ImagePipelineEvaluationReporter
from research_common.image_pipeline_evaluation.image_pipeline_evaluator import ImagePipelineEvaluator
......
......@@ -2,7 +2,8 @@ import logging
from polystar.common.image_pipeline.classifier_image_pipeline import ClassifierImagePipeline
from polystar.common.image_pipeline.models.random_model import RandomModel
from research.armor_digit.armor_digit_pipeline_reporter_factory import ArmorDigitPipelineReporterFactory
from research.robots_at_robots.armor_digit.armor_digit_pipeline_reporter_factory import \
ArmorDigitPipelineReporterFactory
from research_common.dataset.twitch.twitch_roco_datasets import TwitchROCODataset
if __name__ == "__main__":
......
from pathlib import Path
from polystar.common.models.object import Armor
from research.dataset.armor_image_dataset_factory import ArmorImageDatasetGenerator
from research.robots_at_robots.dataset import ArmorImageDatasetGenerator
class ArmorColorDatasetGenerator(ArmorImageDatasetGenerator[str]):
......
......@@ -2,7 +2,7 @@ from pathlib import Path
from typing import Set
from polystar.common.models.object import Armor
from research.dataset.armor_image_dataset_factory import ArmorImageDatasetGenerator
from research.robots_at_robots.dataset import ArmorImageDatasetGenerator
class ArmorDigitDatasetGenerator(ArmorImageDatasetGenerator[int]):
......
import json
from abc import abstractmethod
from pathlib import Path
from typing import TypeVar, Tuple, Iterable
from typing import Iterable, Tuple, TypeVar
import cv2
from polystar.common.models.image import Image
from polystar.common.models.object import Armor
from polystar.common.utils.time import create_time_id
from research.dataset.armor_dataset_factory import ArmorDatasetFactory
from research.robots_at_robots.dataset.armor_dataset_factory import ArmorDatasetFactory
from research_common.dataset.directory_roco_dataset import DirectoryROCODataset
from research_common.image_pipeline_evaluation.image_dataset_generator import ImageDatasetGenerator
......
......@@ -4,7 +4,7 @@ from polystar.common.target_pipeline.objects_validators.confidence_object_valida
from polystar.common.utils.tensorflow import patch_tf_v2
from polystar.common.view.plt_results_viewer import PltResultViewer
from polystar.robots_at_robots.dependency_injection import make_injector
from research.demos.utils import load_tf_model
from research.robots_at_robots.demos.utils import load_tf_model
from research_common.dataset.dji.dji_roco_datasets import DJIROCODataset
from research_common.dataset.split import Split
from research_common.dataset.split_dataset import SplitDataset
......
......@@ -18,7 +18,7 @@ from polystar.common.target_pipeline.target_pipeline import NoTargetFoundExcepti
from polystar.common.utils.tensorflow import patch_tf_v2
from polystar.common.view.plt_results_viewer import PltResultViewer
from polystar.robots_at_robots.dependency_injection import make_injector
from research.demos.utils import load_tf_model
from research.robots_at_robots.demos.utils import load_tf_model
from research_common.dataset.dji.dji_roco_datasets import DJIROCODataset
from research_common.dataset.split import Split
from research_common.dataset.split_dataset import SplitDataset
......
......@@ -5,13 +5,16 @@ import cv2
from tqdm import trange
from polystar.common.models.image import load_images_in_directory
from research.constants import RUNES_DATASET_DIR
from research.robots_at_runes.dataset import gaussian_noise, horizontal_blur
from research.robots_at_runes.dataset.blend import ImageBlender, LabeledImageRotator, LabeledImageScaler
from research.robots_at_runes.constants import RUNES_DATASET_DIR
from research.robots_at_runes.dataset.blend.image_blender import ImageBlender
from research.robots_at_runes.dataset.blend.labeled_image_modifiers.labeled_image_rotator import LabeledImageRotator
from research.robots_at_runes.dataset.blend.labeled_image_modifiers.labeled_image_scaler import LabeledImageScaler
from research.robots_at_runes.dataset.labeled_image import load_labeled_images_in_directory
from research.robots_at_runes.dataset.perturbations.perturbator import ImagePerturbator
from research.robots_at_runes.dataset.perturbations.perturbator_functions.contrast import contrast
from research.robots_at_runes.dataset.perturbations.perturbator_functions.gaussian_blur import gaussian_blur
from research.robots_at_runes.dataset.perturbations.perturbator_functions.gaussian_noise import gaussian_noise
from research.robots_at_runes.dataset.perturbations.perturbator_functions.horizontal_blur import horizontal_blur
class DatasetGenerator:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment