Skip to content

S
seminar-breakout
Commit fb9c4fdf authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix lint

parent 11ca8b2c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/dataflow/dataset/places.py
View file @ fb9c4fdf
...@@ -38,7 +38,7 @@ class Places365Standard(RNGDataFlow): ...@@ -38,7 +38,7 @@ class Places365Standard(RNGDataFlow):
label = line[1] label = line[1]
all_files.append((filepath, label)) all_files.append((filepath, label))
labels.add(label) labels.add(label)
self._labels = sorted(list(labels)) self._labels = sorted(labels)
# class ids are sorted alphabetically: # class ids are sorted alphabetically:
# https://github.com/CSAILVision/places365/blob/master/categories_places365.txt # https://github.com/CSAILVision/places365/blob/master/categories_places365.txt
labelmap = {label: id for id, label in enumerate(self._labels)} labelmap = {label: id for id, label in enumerate(self._labels)}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment