From dfb7647054b8c371689fd50507e9e0c07bd18536 Mon Sep 17 00:00:00 2001
From: deepr <hexiangdong2020@outlook.com>
Date: Wed, 11 Aug 2021 16:33:55 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AE=AD=E7=BB=83=E8=BF=87=E7=A8=8B=E4=B8=AD?=
=?UTF-8?q?=E8=A3=81=E5=88=87=E5=9B=BE=E5=83=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/dataset.py b/dataset.py
index dc28444..c994e54 100644
--- a/dataset.py
+++ b/dataset.py
@@ -23,7 +23,7 @@ class GetDatasetGenerator:
r2 = random.randint(0, 7200 - size2)
image = np.float32(cv2.imread(self.path+"/JPEGImages/"+self.__data[index]+".bmp"))
image = image.swapaxes(1, 2).swapaxes(0, 1)
- image = image[3, r1:r1+size1, r2:r2+size2]
+ image = image[:, r1:r1+size1, r2:r2+size2]
mask = Image.open(self.path+"/SegmentationClass/"+self.__data[index]+".png")
mask = np.int32(np.asarray(mask))
mask = mask[r1:r1+size1, r2:r2+size2]
--
GitLab