Change name of keypointnet inference pipelien

Sharda-Tech · Nov 13, 2019 · becc548 · becc548
1 parent 3ad7b00
commit becc548
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/discovery_of_latent_keypoints/discover_latent_keypoints.py b/examples/discovery_of_latent_keypoints/discover_latent_keypoints.py
@@ -8,7 +8,7 @@
 
 from paz.models import KeypointNetShared
 from paz.models import Projector
-from paz.pipelines import KeypointInference
+from paz.pipelines import KeypointNetInference
 from paz.pipelines import KeypointSharedAugmentation
 from paz.core.sequencer import GeneratingSequencer
 from paz.optimization import KeypointNetLoss
@@ -119,7 +119,7 @@
 log = CSVLogger(os.path.join(save_path, '%s.log' % model_name))
 stop = EarlyStopping('loss', patience=args.stop_patience, verbose=1)
 plateau = ReduceLROnPlateau('loss', patience=args.plateau_patience, verbose=1)
-inferencer = KeypointInference(
+inferencer = KeypointNetInference(
     model.get_layer('keypointnet'), args.num_keypoints)
 images = (sequencer.__getitem__(0)[0][0] * 255).astype('uint8')
 draw = DrawInferences(save_path, images, inferencer)