apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher
2019-03-21 17:26:31 +02:00
parent a024286ec1
commit d047062074
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/datasets.py
+3 -3
View File
@@ -108,7 +108,7 @@ class LoadImagesAndLabels: # for training
def __iter__(self):
self.count = -1
self.shuffled_vector = np.random.permutation(self.nF) if self.augment else np.arange(self.nF)
#self.shuffled_vector = np.random.permutation(self.nF) if self.augment else np.arange(self.nF)
return self
def __getitem__(self, index):
@@ -133,8 +133,8 @@ class LoadImagesAndLabels: # for training
def load_images(self, ia, ib):
img_all, labels_all, img_paths, img_shapes = [], [], [], []
for index, files_index in enumerate(range(ia, ib)):
img_path = self.img_files[self.shuffled_vector[files_index]]
label_path = self.label_files[self.shuffled_vector[files_index]]
img_path = self.img_files[files_index]
label_path = self.label_files[files_index]
img = cv2.imread(img_path) # BGR
assert img is not None, 'File Not Found ' + img_path