Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
  • Loading branch information
MarcBS committed Jul 5, 2017
2 parents e85d721 + 18b98f1 commit 1a51cbf
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions keras_wrapper/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -2741,7 +2741,6 @@ def convert_GT_3DLabels_to_bboxes(self, gt):
# ------------------------------------------------------- #

def preprocessImages(self, path_list, id, set_name, img_size, img_size_crop, use_RGB):

if isinstance(path_list, str) and os.path.isfile(path_list): # path to list of images' paths
data = []
with open(path_list, 'r') as list_:
Expand Down Expand Up @@ -3222,8 +3221,8 @@ def getXY(self, set_name, k, normalization_type='0-1', normalization=False, mean
if not debug:
if type_out == 'categorical':
nClasses = len(self.dic_classes[id_out])
load_sample_weights = self.sample_weights[id_out][set_name]
y = self.loadCategorical(y, nClasses, id_out, load_sample_weights)
#load_sample_weights = self.sample_weights[id_out][set_name]
y = self.loadCategorical(y, nClasses)
elif type_out == 'binary':
y = self.loadBinary(y, id_out)
elif type_out == 'real':
Expand Down

0 comments on commit 1a51cbf

Please sign in to comment.