updates
This commit is contained in:
parent
658ef8a272
commit
fd802910ec
|
@ -285,7 +285,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
|
||||||
h, w, _ = img.shape
|
h, w, _ = img.shape
|
||||||
img = cv2.resize(img, (int(w * r), int(h * r)), interpolation=cv2.INTER_LINEAR) # INTER_LINEAR fastest
|
img = cv2.resize(img, (int(w * r), int(h * r)), interpolation=cv2.INTER_LINEAR) # INTER_LINEAR fastest
|
||||||
|
|
||||||
if self.n < 3000: # cache into memory if image count < 3000
|
if self.n < 5000: # cache into memory if image count < 5000
|
||||||
self.imgs[index] = img
|
self.imgs[index] = img
|
||||||
|
|
||||||
# Augment colorspace
|
# Augment colorspace
|
||||||
|
|
Loading…
Reference in New Issue