From ea73386e5a21f6b6d4f2bdc0ba1f9f8a7ced3f2a Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 15 Nov 2022 14:19:47 +0100
Subject: [PATCH] Add Copy-Paste to `mosaic9()` (#10165)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
---
 utils/dataloaders.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/utils/dataloaders.py b/utils/dataloaders.py
index 621c03cd2..54d3f7bbb 100644
--- a/utils/dataloaders.py
+++ b/utils/dataloaders.py
@@ -869,6 +869,7 @@ class LoadImagesAndLabels(Dataset):
         # 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,