From 09dfbd63af2e1d47be4634cb4432f508e92ae86f Mon Sep 17 00:00:00 2001
From: deepr <hexiangdong2020@outlook.com>
Date: Wed, 11 Aug 2021 16:31:35 +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 | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/dataset.py b/dataset.py
index dc91831..dc28444 100644
--- a/dataset.py
+++ b/dataset.py
@@ -3,6 +3,7 @@ import os
from PIL import Image
import cv2
import mindspore.dataset as ds
+import random
class GetDatasetGenerator:
@@ -16,12 +17,16 @@ class GetDatasetGenerator:
line = f.readline() # 璇诲彇涓嬩竴琛�
def __getitem__(self, index):
+ size1 = 3400
+ size2 = 3600
+ r1 = random.randint(0, 6800 - size1)
+ 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]
mask = Image.open(self.path+"/SegmentationClass/"+self.__data[index]+".png")
mask = np.int32(np.asarray(mask))
- print(image.shape)
- print(mask.shape)
+ mask = mask[r1:r1+size1, r2:r2+size2]
return image, mask
def __len__(self):
--
GitLab