scribble-annotation-generator 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

scribble_annotation_generator/gmm_random_generator.py

@@ -54,11 +54,13 @@ class GMMRandomGenerator:
         num_components: int = 5,
         random_state: int = 0,
         class_names: Optional[List[str]] = None,
+        colour_map: Optional[dict[tuple[int, int, int], int]] = None,
     ) -> None:
         self.num_classes = num_classes
         self.num_components = num_components
         self.random_state = random_state
         self.class_names = class_names or [f"class_{i}" for i in range(num_classes)]
+        self.colour_map = colour_map
 
         self.class_gmms: Dict[int, GaussianMixture] = {}
         self.class_stats: Dict[int, Dict[str, float]] = {}
@@ -193,7 +195,6 @@ class GMMRandomGenerator:
     def sample_image(
         self,
         image_shape: Tuple[int, int],
-        colour_map: Optional[dict[tuple[int, int, int], int]] = None,
         overlap_iters: int = 40,
         overlap_margin: float = 0.05,
     ) -> Tuple[np.ndarray, str]:
@@ -220,7 +221,7 @@ class GMMRandomGenerator:
             image_shape=image_shape,
             objects=objects,
             classes=classes,
-            colour_map=colour_map,
+            colour_map=self.colour_map,
         )
 
         # Generate description string
@@ -579,6 +580,7 @@ def train_gmm_model(data_dir: str) -> GMMRandomGenerator:
         num_classes=num_classes,
         num_components=5,
         class_names=class_names,
+        colour_map=colour_map,
     ).fit(dataset)
 
     return generator
@@ -617,6 +619,7 @@ def main(argv: Optional[List[str]] = None) -> None:
         num_classes=args.num_classes,
         num_components=args.num_components,
         class_names=class_names,
+        colour_map=colour_map,
     ).fit(dataset)
     print()
 
@@ -626,7 +629,6 @@ def main(argv: Optional[List[str]] = None) -> None:
     for idx in range(args.num_samples):
         image, description = generator.sample_image(
             image_shape=(args.height, args.width),
-            colour_map=colour_map,
             overlap_iters=args.overlap_iters,
             overlap_margin=args.overlap_margin,
         )

{scribble_annotation_generator-0.1.0.dist-info → scribble_annotation_generator-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: scribble-annotation-generator
-Version: 0.1.0
+Version: 0.1.1
 Summary: Programmatically generate semi-realistic synthetic scribble annotations based on statistics from existing scribble datasets
 Project-URL: Homepage, https://github.com/alexsenden/scribble-annotation-generator
 Project-URL: Repository, https://github.com/alexsenden/scribble-annotation-generator

{scribble_annotation_generator-0.1.0.dist-info → scribble_annotation_generator-0.1.1.dist-info}/RECORD

@@ -3,10 +3,10 @@ scribble_annotation_generator/cli.py,sha256=e6Ufmf0CLMrKZCL08pB5RqiiQMNJBU9uZwcL
 scribble_annotation_generator/crop_field.py,sha256=zh0jNpjz5_pWzybQNfga9hYqRIYPXysMPp_xNvL37gM,10774
 scribble_annotation_generator/dataset.py,sha256=jauKr8ZBJ1o8jEn8T_RKpVgqu8kwLmucIMyhZCkPiTg,3298
 scribble_annotation_generator/debug.py,sha256=YJnfkBJL7Vwlqz9SWeybAhwID8Pcwzp_RFnL9xPOQyI,1194
-scribble_annotation_generator/gmm_random_generator.py,sha256=j7ZLEYTH8COJsKFJXHBzY4qKTEreDDONmmbad3xaBi8,22161
+scribble_annotation_generator/gmm_random_generator.py,sha256=TFcjFAB9gpZdVkCm0p7H_vNVF2NrXmRFonkH0NXyZPY,22230
 scribble_annotation_generator/nn.py,sha256=aSQPkVpvsya942hz02LKoUEkZfL_f7lCvYQ5cI8R3Ts,16627
 scribble_annotation_generator/utils.py,sha256=gluwQSroMd4bg6iwchiv4VBTK57t4OvScH8uv29erWY,13628
-scribble_annotation_generator-0.1.0.dist-info/METADATA,sha256=p4MOjTbGTeAi5zS8HGXGj5pK5Op5vE5YCM3JDhRAuIY,4309
-scribble_annotation_generator-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-scribble_annotation_generator-0.1.0.dist-info/entry_points.txt,sha256=A5UbznzAcE5XF5MZrth2rdLvG2IQXhjK2lhklUf9QyU,89
-scribble_annotation_generator-0.1.0.dist-info/RECORD,,
+scribble_annotation_generator-0.1.1.dist-info/METADATA,sha256=n5U2p5kcY9Jp8uIPU5f-oP7ycP5g4puJ0t3Bbw2pGb4,4309
+scribble_annotation_generator-0.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+scribble_annotation_generator-0.1.1.dist-info/entry_points.txt,sha256=A5UbznzAcE5XF5MZrth2rdLvG2IQXhjK2lhklUf9QyU,89
+scribble_annotation_generator-0.1.1.dist-info/RECORD,,