edit data format in load_bin

plsc/utils/jpeg_reader.py

@@ -252,7 +252,11 @@ def load_bin(path, image_size, data_format ='NCHW'):
         bins, issame_list = pickle.load(open(path, 'rb'), encoding='bytes')
     data_list = []
     for flip in [0, 1]:
-        data = np.empty((len(issame_list) * 2, 3, image_size[0], image_size[1]))
+        if data_format == 'NCHW':
+            data = np.empty((len(issame_list) * 2, 3, image_size[0], image_size[1]))
+        else:
+            #NHWC
+            data = np.empty((len(issame_list) * 2, image_size[0], image_size[1], 3))
         data_list.append(data)
     for i in range(len(issame_list) * 2):
         _bin = bins[i]