Commit fb9c4fdf authored by Yuxin Wu's avatar Yuxin Wu

fix lint

parent 11ca8b2c
......@@ -38,7 +38,7 @@ class Places365Standard(RNGDataFlow):
label = line[1]
all_files.append((filepath, label))
labels.add(label)
self._labels = sorted(list(labels))
self._labels = sorted(labels)
# class ids are sorted alphabetically:
# https://github.com/CSAILVision/places365/blob/master/categories_places365.txt
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