From 04cb07fc2d169121465f0b6147fedc4ca2f82fa9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 5 Mar 2021 13:04:47 -0800 Subject: [PATCH] Update datasets.py --- utils/datasets.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/datasets.py b/utils/datasets.py index ea84bd1de05b..d6ab16518034 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -666,7 +666,7 @@ def load_mosaic(self, index): labels4, segments4 = [], [] s = self.img_size yc, xc = [int(random.uniform(-x, 2 * s + x)) for x in self.mosaic_border] # mosaic center x, y - indices = [index] + [random.randint(0, self.n - 1) for _ in range(3)] # 3 additional image indices + indices = [index] + [self.indices[random.randint(0, self.n - 1)] for _ in range(3)] # 3 additional image indices for i, index in enumerate(indices): # Load image img, _, (h, w) = load_image(self, index) @@ -721,7 +721,7 @@ def load_mosaic9(self, index): labels9, segments9 = [], [] s = self.img_size - indices = [index] + [random.randint(0, self.n - 1) for _ in range(8)] # 8 additional image indices + indices = [index] + [self.indices[random.randint(0, self.n - 1)] for _ in range(8)] # 8 additional image indices for i, index in enumerate(indices): # Load image img, _, (h, w) = load_image(self, index)