Formatting fixes

Jerry-robot · Jul 15, 2020 · 75c939b · 75c939b
1 parent adfd6f3
commit 75c939b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/run_offline.py b/run_offline.py
@@ -16,7 +16,7 @@
 
 def parse_args():
     parser = argparse.ArgumentParser(description='Evaluate network')
-    parser.add_argument('--network', type=str, default='cornell_rgbd_iou_0.95',
+    parser.add_argument('--network', type=str,
                         help='Path to saved network to evaluate')
     parser.add_argument('--rgb_path', type=str, default='cornell/08/pcd0845r.png',
                         help='RGB Image path')

diff --git a/train_network.py b/train_network.py
@@ -116,7 +116,8 @@ def validate(net, device, val_data):
             q_out, ang_out, w_out = post_process_output(lossd['pred']['pos'], lossd['pred']['cos'],
                                                         lossd['pred']['sin'], lossd['pred']['width'])
 
-            s = evaluation.calculate_iou_match(q_out, ang_out,
+            s = evaluation.calculate_iou_match(q_out,
+                                               ang_out,
                                                val_data.dataset.get_gtbb(didx, rot, zoom_factor),
                                                no_grasps=1,
                                                grasp_width=w_out,

diff --git a/utils/dataset_processing/grasp.py b/utils/dataset_processing/grasp.py
@@ -415,8 +415,8 @@ def to_jacquard(self, scale=1):
         """
         # Output in jacquard format.
         return '%0.2f;%0.2f;%0.2f;%0.2f;%0.2f' % (
-        self.center[1] * scale, self.center[0] * scale, -1 * self.angle * 180 / np.pi, self.length * scale,
-        self.width * scale)
+            self.center[1] * scale, self.center[0] * scale, -1 * self.angle * 180 / np.pi, self.length * scale,
+            self.width * scale)
 
 
 def detect_grasps(q_img, ang_img, width_img=None, no_grasps=1):