Fix bug with automatic keypoint checker

Sharda-Tech · Nov 8, 2019 · 566ddf1 · 566ddf1
1 parent 06ccc8e
commit 566ddf1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/discovery_of_latent_keypoints/discover_latent_keypoints.py b/examples/discovery_of_latent_keypoints/discover_latent_keypoints.py
@@ -119,7 +119,8 @@
 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(model.get_layer('keypointnet'))
+inferencer = KeypointInference(
+    model.get_layer('keypointnet'), args.num_keypoints)
 images = (sequencer.__getitem__(0)[0][0] * 255).astype('uint8')
 draw = DrawInferences(save_path, images, inferencer)
 model_path = os.path.join(save_path, '%s_weights.hdf5' % model_name)