From a82a60c2e683b8817cbb90728a2540defb388d09 Mon Sep 17 00:00:00 2001
From: Mathieu Beligon <mathieu@feedly.com>
Date: Mon, 22 Mar 2021 12:09:00 -0400
Subject: [PATCH] [twitch] (masks) move to masks dir
---
src/research/dataset/twitch/mask_detector.py | 26 ++++++------------
.../twitch/{ => masks}/mask_aerial.jpg | Bin
.../{ => masks}/mask_aerial_blue_2_hd.jpg | Bin
.../{ => masks}/mask_aerial_blue_hd.jpg | Bin
.../{ => masks}/mask_aerial_red_2_hd.jpg | Bin
.../twitch/{ => masks}/mask_aerial_red_hd.jpg | Bin
.../dataset/twitch/{ => masks}/mask_bonus.jpg | Bin
.../twitch/{ => masks}/mask_bonus_2.jpg | Bin
.../twitch/{ => masks}/mask_bonus_3.jpg | Bin
.../twitch/{ => masks}/mask_robot_view.jpg | Bin
.../twitch/{ => masks}/mask_robot_view_hd.jpg | Bin
src/research/dataset/twitch/resize_mask.py | 7 -----
12 files changed, 9 insertions(+), 24 deletions(-)
rename src/research/dataset/twitch/{ => masks}/mask_aerial.jpg (100%)
rename src/research/dataset/twitch/{ => masks}/mask_aerial_blue_2_hd.jpg (100%)
rename src/research/dataset/twitch/{ => masks}/mask_aerial_blue_hd.jpg (100%)
rename src/research/dataset/twitch/{ => masks}/mask_aerial_red_2_hd.jpg (100%)
rename src/research/dataset/twitch/{ => masks}/mask_aerial_red_hd.jpg (100%)
rename src/research/dataset/twitch/{ => masks}/mask_bonus.jpg (100%)
rename src/research/dataset/twitch/{ => masks}/mask_bonus_2.jpg (100%)
rename src/research/dataset/twitch/{ => masks}/mask_bonus_3.jpg (100%)
rename src/research/dataset/twitch/{ => masks}/mask_robot_view.jpg (100%)
rename src/research/dataset/twitch/{ => masks}/mask_robot_view_hd.jpg (100%)
delete mode 100644 src/research/dataset/twitch/resize_mask.py
diff --git a/src/research/dataset/twitch/mask_detector.py b/src/research/dataset/twitch/mask_detector.py
index 4d63b52..111aa60 100644
--- a/src/research/dataset/twitch/mask_detector.py
+++ b/src/research/dataset/twitch/mask_detector.py
@@ -4,7 +4,7 @@ import numpy as np
from polystar.models.image import Image, load_image
-DIR_PATH = Path(__file__).parent
+MASKS_DIR = Path(__file__).parent / "masks"
class Mask:
@@ -20,14 +20,14 @@ class Mask:
return value <= self._threshold
-robot_view_mask_hd = Mask(DIR_PATH / "mask_robot_view_hd.jpg", 20)
-aerial_view_mask_red_hd = Mask(DIR_PATH / "mask_aerial_red_hd.jpg", 15)
-aerial_view_mask_red_2_hd = Mask(DIR_PATH / "mask_aerial_red_2_hd.jpg", 15)
-aerial_view_mask_blue_hd = Mask(DIR_PATH / "mask_aerial_blue_hd.jpg", 15)
-aerial_view_mask_blue_2_hd = Mask(DIR_PATH / "mask_aerial_blue_2_hd.jpg", 15)
-bonus_view_mask_hd = Mask(DIR_PATH / "mask_bonus.jpg", 20)
-bonus_2_view_mask_hd = Mask(DIR_PATH / "mask_bonus_2.jpg", 20)
-bonus_3_view_mask_hd = Mask(DIR_PATH / "mask_bonus_3.jpg", 20)
+robot_view_mask_hd = Mask(MASKS_DIR / "mask_robot_view_hd.jpg", 20)
+aerial_view_mask_red_hd = Mask(MASKS_DIR / "mask_aerial_red_hd.jpg", 15)
+aerial_view_mask_red_2_hd = Mask(MASKS_DIR / "mask_aerial_red_2_hd.jpg", 15)
+aerial_view_mask_blue_hd = Mask(MASKS_DIR / "mask_aerial_blue_hd.jpg", 15)
+aerial_view_mask_blue_2_hd = Mask(MASKS_DIR / "mask_aerial_blue_2_hd.jpg", 15)
+bonus_view_mask_hd = Mask(MASKS_DIR / "mask_bonus.jpg", 20)
+bonus_2_view_mask_hd = Mask(MASKS_DIR / "mask_bonus_2.jpg", 20)
+bonus_3_view_mask_hd = Mask(MASKS_DIR / "mask_bonus_3.jpg", 20)
def is_aerial_view(image: Image) -> bool:
@@ -41,11 +41,3 @@ def is_aerial_view(image: Image) -> bool:
def has_bonus_icon(image: Image) -> bool:
return bonus_view_mask_hd.match(image) or bonus_2_view_mask_hd.match(image) or bonus_3_view_mask_hd.match(image)
-
-
-if __name__ == "__main__":
- has_bonus_icon(
- load_image(
- Path("/Users/cytadel/polystar/cv-code/dataset/twitch/robots-views/470152932/470152932-frame-007460.jpg")
- )
- )
diff --git a/src/research/dataset/twitch/mask_aerial.jpg b/src/research/dataset/twitch/masks/mask_aerial.jpg
similarity index 100%
rename from src/research/dataset/twitch/mask_aerial.jpg
rename to src/research/dataset/twitch/masks/mask_aerial.jpg
diff --git a/src/research/dataset/twitch/mask_aerial_blue_2_hd.jpg b/src/research/dataset/twitch/masks/mask_aerial_blue_2_hd.jpg
similarity index 100%
rename from src/research/dataset/twitch/mask_aerial_blue_2_hd.jpg
rename to src/research/dataset/twitch/masks/mask_aerial_blue_2_hd.jpg
diff --git a/src/research/dataset/twitch/mask_aerial_blue_hd.jpg b/src/research/dataset/twitch/masks/mask_aerial_blue_hd.jpg
similarity index 100%
rename from src/research/dataset/twitch/mask_aerial_blue_hd.jpg
rename to src/research/dataset/twitch/masks/mask_aerial_blue_hd.jpg
diff --git a/src/research/dataset/twitch/mask_aerial_red_2_hd.jpg b/src/research/dataset/twitch/masks/mask_aerial_red_2_hd.jpg
similarity index 100%
rename from src/research/dataset/twitch/mask_aerial_red_2_hd.jpg
rename to src/research/dataset/twitch/masks/mask_aerial_red_2_hd.jpg
diff --git a/src/research/dataset/twitch/mask_aerial_red_hd.jpg b/src/research/dataset/twitch/masks/mask_aerial_red_hd.jpg
similarity index 100%
rename from src/research/dataset/twitch/mask_aerial_red_hd.jpg
rename to src/research/dataset/twitch/masks/mask_aerial_red_hd.jpg
diff --git a/src/research/dataset/twitch/mask_bonus.jpg b/src/research/dataset/twitch/masks/mask_bonus.jpg
similarity index 100%
rename from src/research/dataset/twitch/mask_bonus.jpg
rename to src/research/dataset/twitch/masks/mask_bonus.jpg
diff --git a/src/research/dataset/twitch/mask_bonus_2.jpg b/src/research/dataset/twitch/masks/mask_bonus_2.jpg
similarity index 100%
rename from src/research/dataset/twitch/mask_bonus_2.jpg
rename to src/research/dataset/twitch/masks/mask_bonus_2.jpg
diff --git a/src/research/dataset/twitch/mask_bonus_3.jpg b/src/research/dataset/twitch/masks/mask_bonus_3.jpg
similarity index 100%
rename from src/research/dataset/twitch/mask_bonus_3.jpg
rename to src/research/dataset/twitch/masks/mask_bonus_3.jpg
diff --git a/src/research/dataset/twitch/mask_robot_view.jpg b/src/research/dataset/twitch/masks/mask_robot_view.jpg
similarity index 100%
rename from src/research/dataset/twitch/mask_robot_view.jpg
rename to src/research/dataset/twitch/masks/mask_robot_view.jpg
diff --git a/src/research/dataset/twitch/mask_robot_view_hd.jpg b/src/research/dataset/twitch/masks/mask_robot_view_hd.jpg
similarity index 100%
rename from src/research/dataset/twitch/mask_robot_view_hd.jpg
rename to src/research/dataset/twitch/masks/mask_robot_view_hd.jpg
diff --git a/src/research/dataset/twitch/resize_mask.py b/src/research/dataset/twitch/resize_mask.py
deleted file mode 100644
index 2386299..0000000
--- a/src/research/dataset/twitch/resize_mask.py
+++ /dev/null
@@ -1,7 +0,0 @@
-from pathlib import Path
-
-from imutils import resize
-
-from polystar.models.image import load_image, save_image
-
-save_image(resize(load_image(Path("mask_aerial.jpg")), 1920, 1080), Path("mask_aerial_red_hd.jpg"))
--
GitLab