[common] (refactor) Move research into research/common

common/README.md

@@ -68,13 +68,13 @@ You have 2 options to download the videos:
 1. First, go on the [google sheet](https://docs.google.com/spreadsheets/d/1kIrMOjcKJ8hslZoVMx1D0H7QYj9nQLFvzUAQ1U4Le-I/edit#gid=0), and choose a video that nobody already did, and put your name in the 2nd column
 2. Download the video, in **720p**
 3. Rename it using the video id on twitch, and place it in [../dataset/twitch/videos](../dataset/twitch/videos)
-5. Launch the python script [./research_common/scripts/extract_robots_views_from_video.py](research_common/scripts/extract_robots_views_from_video.py), with the video id as parameter (In Pycharm, `Run` > `Edit Configurations...`, then in parameters enter the id). You can put multiple video ids by separating them with spaces.
+5. Launch the python script [./research/common/scripts/extract_robots_views_from_video.py](research/common/scripts/extract_robots_views_from_video.py), with the video id as parameter (In Pycharm, `Run` > `Edit Configurations...`, then in parameters enter the id). You can put multiple video ids by separating them with spaces.
 
 The frames will appear in the [../dataset/twitch/robots-views](../dataset/twitch/robots-views) folder.
 
 #### Aerial dataset
 
-Once you have the robots views in the [../dataset/twitch/robots-views](../dataset/twitch/robots-views) folder, run the python script [./research_common/scripts/move_aerial_views.py](research_common/scripts/move_aerial_views.py). It will put the aerial views in the [../dataset/twitch/aerial-views](../dataset/twitch/aerial-views) directory.
+Once you have the robots views in the [../dataset/twitch/robots-views](../dataset/twitch/robots-views) folder, run the python script [./research/common/scripts/move_aerial_views.py](research/common/scripts/move_aerial_views.py). It will put the aerial views in the [../dataset/twitch/aerial-views](../dataset/twitch/aerial-views) directory.
 
 
 

common/research_common/dataset/directory_roco_dataset.py → common/research/common/dataset/directory_roco_dataset.py

 from pathlib import Path
 from typing import Iterable
 
-from research_common.dataset.roco_dataset import ROCODataset
+from research.common.dataset.roco_dataset import ROCODataset
 
 
 class DirectoryROCODataset(ROCODataset):

common/research_common/dataset/dji/dji_roco_datasets.py → common/research/common/dataset/dji/dji_roco_datasets.py

 from enum import Enum
 
-from research_common.constants import DJI_ROCO_DSET_DIR
-from research_common.dataset.directory_roco_dataset import DirectoryROCODataset
+from research.common.constants import DJI_ROCO_DSET_DIR
+from research.common.dataset.directory_roco_dataset import DirectoryROCODataset
 
 
 class DJIROCODataset(DirectoryROCODataset, Enum):

common/research_common/dataset/dji/dji_roco_zoomed_datasets.py → common/research/common/dataset/dji/dji_roco_zoomed_datasets.py

 from enum import Enum, auto
 
 from polystar.common.utils.str_utils import camel2snake
-from research_common.constants import DJI_ROCO_ZOOMED_DSET_DIR
-from research_common.dataset.directory_roco_dataset import DirectoryROCODataset
+from research.common.constants import DJI_ROCO_ZOOMED_DSET_DIR
+from research.common.dataset.directory_roco_dataset import DirectoryROCODataset
 
 
 class DJIROCOZoomedDataset(DirectoryROCODataset, Enum):

common/research_common/dataset/improvement/zoom.py → common/research/common/dataset/improvement/zoom.py

@@ -7,7 +7,7 @@ from polystar.common.models.box import Box
 from polystar.common.models.image_annotation import ImageAnnotation
 from polystar.common.target_pipeline.objects_validators.in_box_validator import InBoxValidator
 from polystar.common.view.plt_results_viewer import PltResultViewer
-from research_common.dataset.dji.dji_roco_datasets import DJIROCODataset
+from research.common.dataset.dji.dji_roco_datasets import DJIROCODataset
 
 
 def crop_image_annotation(image_annotation: ImageAnnotation, box: Box, min_coverage: float) -> ImageAnnotation:

common/research_common/dataset/roco_dataset_descriptor.py → common/research/common/dataset/roco_dataset_descriptor.py

@@ -7,10 +7,10 @@ from pandas import DataFrame
 
 from polystar.common.models.object import Armor, ObjectType
 from polystar.common.utils.markdown import MarkdownFile
-from research_common.dataset.dji.dji_roco_datasets import DJIROCODataset
-from research_common.dataset.dji.dji_roco_zoomed_datasets import DJIROCOZoomedDataset
-from research_common.dataset.roco_dataset import ROCODataset
-from research_common.dataset.twitch.twitch_roco_datasets import TwitchROCODataset
+from research.common.dataset.dji.dji_roco_datasets import DJIROCODataset
+from research.common.dataset.dji.dji_roco_zoomed_datasets import DJIROCOZoomedDataset
+from research.common.dataset.roco_dataset import ROCODataset
+from research.common.dataset.twitch.twitch_roco_datasets import TwitchROCODataset
 
 
 @dataclass

common/research_common/dataset/split.py → common/research/common/dataset/split.py

 from enum import Enum
 from pathlib import Path
 
-from research_common.dataset.directory_roco_dataset import DirectoryROCODataset
+from research.common.dataset.directory_roco_dataset import DirectoryROCODataset
 
 
 class Split(Enum):

common/research_common/dataset/split_dataset.py → common/research/common/dataset/split_dataset.py

 from pathlib import Path
 from typing import Iterable, List
 
-from research_common.dataset.directory_roco_dataset import DirectoryROCODataset
-from research_common.dataset.dji.dji_roco_datasets import DJIROCODataset
-from research_common.dataset.split import Split
+from research.common.dataset.directory_roco_dataset import DirectoryROCODataset
+from research.common.dataset.dji.dji_roco_datasets import DJIROCODataset
+from research.common.dataset.split import Split
 
 
 class SplitDataset(DirectoryROCODataset):

common/research_common/dataset/tensorflow_record.py → common/research/common/dataset/tensorflow_record.py

@@ -8,8 +8,8 @@ from tqdm import tqdm
 
 from polystar.common.models.image_annotation import ImageAnnotation
 from polystar.common.models.label_map import label_map
-from research_common.constants import TENSORFLOW_RECORDS_DIR
-from research_common.dataset.roco_dataset import ROCODataset
+from research.common.constants import TENSORFLOW_RECORDS_DIR
+from research.common.dataset.roco_dataset import ROCODataset
 
 
 class TensorflowRecordFactory:

common/research_common/dataset/twitch/aerial_view_detector.py → common/research/common/dataset/twitch/aerial_view_detector.py

@@ -2,8 +2,8 @@ from pathlib import Path
 
 from skimage import io
 
-from research_common.constants import TWITCH_DSET_DIR
-from research_common.dataset.twitch.mask_detector import MaskDetector
+from research.common.constants import TWITCH_DSET_DIR
+from research.common.dataset.twitch.mask_detector import MaskDetector
 
 aerial_view_detector = MaskDetector(
     Path(__file__).parent / "mask_aerial.jpg",