DenseNet161 with SGD

ajing · Jun 14, 2017 · ffd58e1 · ffd58e1
1 parent 8b5a333
commit ffd58e1
Show file tree

Hide file tree

Showing 5 changed files with 240 additions and 111 deletions.
diff --git a/data/kgdataset.py b/data/kgdataset.py
@@ -98,12 +98,12 @@ def __len__(self):
         return len(self.images)
 
 
-
 def randomVerticalFlip(img, u=0.5):
     if random.random() < u:
-        img = cv2.flip(img,0)  #np.flipud(img)  #cv2.flip(img,0) ##up-down
+        img = cv2.flip(img, 0)  #np.flipud(img)  #cv2.flip(img,0) ##up-down
     return img
 
+
 def randomHorizontalFlip(img, u=0.5):
     shape=img.shape
     if random.random() < u: