Update viz_eval.py

charlesq34 · Oct 1, 2018 · 2ffdd34 · 2ffdd34
1 parent 6d4646f
commit 2ffdd34
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sunrgbd/sunrgbd_detection/viz_eval.py b/sunrgbd/sunrgbd_detection/viz_eval.py
@@ -16,6 +16,7 @@
 from sunrgbd_data import sunrgbd_object
 from utils import load_zipped_pickle
 sys.path.append(os.path.join(BASE_DIR, '../../train'))
+sys.path.append(os.path.join(BASE_DIR, '../../mayavi'))
 from box_util import box3d_iou
 
 parser = argparse.ArgumentParser()
@@ -31,7 +32,7 @@
 VISU = FLAGS.viz
 if VISU:
     import mayavi.mlab as mlab
-    from view_pc import draw_lidar, draw_gt_boxes3d
+    from viz_util import draw_lidar, draw_gt_boxes3d
 
 #with open(FLAGS.result_path, 'rb') as fp:
 #    ps_list = pickle.load(fp)