update code

chainer · Aug 20, 2017 · cfcccc7 · cfcccc7
1 parent 26231db
commit cfcccc7
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 263 deletions.
diff --git a/chainercv/links/model/feature_extraction_predictor.py b/chainercv/links/model/feature_extraction_predictor.py
diff --git a/docs/source/reference/links.rst b/docs/source/reference/links.rst
@@ -18,7 +18,7 @@ Feature extraction models can be used to extract feature(s) given images.
 .. autoclass:: chainercv.links.SequentialFeatureExtractor
    :members:
 
-.. autoclass:: chainercv.links.FeatureExtractionPredictor
+.. autoclass:: chainercv.links.FeaturePredictor
 
 
 Detection

diff --git a/examples/classification/eval_imagenet.py b/examples/classification/eval_imagenet.py
@@ -10,7 +10,7 @@
 
 from chainercv.datasets import directory_parsing_label_names
 from chainercv.datasets import DirectoryParsingClassificationDataset
-from chainercv.links import FeatureExtractionPredictor
+from chainercv.links import FeaturePredictor
 from chainercv.links import VGG16
 
 from chainercv.utils import apply_prediction_to_iterator
@@ -59,7 +59,7 @@ def main():
         else:
             extractor = VGG16(pretrained_model='imagenet',
                               n_class=len(label_names))
-    model = FeatureExtractionPredictor(extractor, crop=args.crop)
+    model = FeaturePredictor(extractor, 224, crop=args.crop)
 
     if args.gpu >= 0:
         chainer.cuda.get_device(args.gpu).use()

diff --git a/tests/links_tests/model_tests/test_feature_extraction_predictor.py b/tests/links_tests/model_tests/test_feature_extraction_predictor.py