Add Copy-Paste to mosaic9() (#10165)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
ultralytics · Nov 15, 2022 · ea73386 · ea73386
1 parent abbfd69
commit ea73386
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/dataloaders.py b/utils/dataloaders.py
@@ -869,6 +869,7 @@ def load_mosaic9(self, index):
  # img9, labels9 = replicate(img9, labels9) # replicate
 
  # Augment
+ img9, labels9, segments9 = copy_paste(img9, labels9, segments9, p=self.hyp['copy_paste'])
  img9, labels9 = random_perspective(img9,
  labels9,
  segments9,