Add parameters to argparse for experiments

Sharda-Tech · Dec 11, 2019 · da5ba6c · da5ba6c
1 parent 0e0b9f1
commit da5ba6c
Showing 1 changed file with 7 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
@@ -75,6 +75,10 @@
                     help='Light intensity from poseur')
 parser.add_argument('-bk', '--background', default=0, type=int,
                     help='Background color')
+parser.add_argument('-rn', '--rotation_noise', default=0.1, type=float,
+                    help='Sigma of noise added to relative pose estimation')
+parser.add_argument('-sd', '--separation_delta', default=0.05, type=float,
+                    help='Sigma of noise added to relative pose estimation')
 parser.add_argument('-ap', '--alpha', default=0.1, type=float,
                     help='Alpha leaky-relu parameter')
 parser.add_argument('-nm', '--num_mean_samples', default=1000, type=int,
@@ -105,8 +109,9 @@
 
 # loss instantiation
 name = ['consistency', 'silhouette', 'separation', 'relative_pose', 'variance']
-weights = dict(zip(name, args.loss_weights))
-loss = KeypointNetLoss(args.num_keypoints, focal_length, loss_weights=weights)
+loss_weights = dict(zip(name, args.loss_weights))
+loss = KeypointNetLoss(args.num_keypoints, focal_length, args.rotation_noise,
+                       args.separation_delta, loss_weights)
 losses = {'uvz_points-shared': loss.uvz_points,
           'uv_volumes-shared': loss.uv_volumes}