Update dataset.py

chao-tan · Aug 7, 2020 · db5ed47 · db5ed47
1 parent 5d007fc
commit db5ed47
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dataset.py b/dataset.py
@@ -161,10 +161,10 @@ def next_batch(self,batchsize):
 
         for i in range(batchsize):
             for k in range(self.seq_length):
-                x_batch[i,k,0,:,:] = cv2.imread(root_path+"/SCMD_"+str(random_order[i])+"/"+str(k+1)+".png",cv2.IMREAD_GRAYSCALE)
-            y_batch[i,0,0,:,:] = cv2.imread(root_path+"/SCMD_"+str(random_order[i])+"/"+str(self.seq_length+1)+".png",cv2.IMREAD_GRAYSCALE)
+                x_batch[i,k,0,:,:] = cv2.imread(root_path+"/"+str(random_order[i])+"/"+str(k+1)+".png",cv2.IMREAD_GRAYSCALE)
+            y_batch[i,0,0,:,:] = cv2.imread(root_path+"/"+str(random_order[i])+"/"+str(self.seq_length+1)+".png",cv2.IMREAD_GRAYSCALE)
 
-        x_batch,y_batch = x_batch*10.0,y_batch*10.0
+        x_batch,y_batch = x_batch,y_batch
 
         if self.baseline:
             x_batch = x_batch[:,:,:,100,100]