Skip to content
This repository has been archived by the owner on Jul 2, 2021. It is now read-only.

Use LabelDataset naming convention #405

Merged
merged 10 commits into from
Sep 28, 2017
Merged
Prev Previous commit
Next Next commit
update eval_imagenet
  • Loading branch information
yuyu2172 committed Aug 21, 2017
commit f7eb7f363c1f66a4d99b34767ce9883aac9db3ac
4 changes: 2 additions & 2 deletions examples/classification/eval_imagenet.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
from chainer import iterators

from chainercv.datasets import directory_parsing_label_names
from chainercv.datasets import DirectoryParsingClassificationDataset
from chainercv.datasets import DirectoryParsingLabelDataset
from chainercv.links import FeaturePredictor
from chainercv.links import VGG16

Expand Down Expand Up @@ -45,7 +45,7 @@ def main():
parser.add_argument('--crop', choices=('center', '10'), default='center')
args = parser.parse_args()

dataset = DirectoryParsingClassificationDataset(args.val)
dataset = DirectoryParsingLabelDataset(args.val)
label_names = directory_parsing_label_names(args.val)
iterator = iterators.MultiprocessIterator(
dataset, args.batchsize, repeat=False, shuffle=False,
Expand Down