diff --git a/src/research/dataset/twitch/mask_detector.py b/src/research/dataset/twitch/mask_detector.py
index 4d63b526c8dda29406ba90f826450a5440f6b773..111aa609ea8af63bf403f5e456e9b4c9c19e6b49 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 238629956b3c4f23043b34287016a14f9b7c5c57..0000000000000000000000000000000000000000
--- 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"))