From dc656c05da85e9e732c56a887bb665e1b2a52289 Mon Sep 17 00:00:00 2001
From: NoOne <365867978@qq.com>
Date: Fri, 27 May 2022 18:56:10 -0400
Subject: [PATCH] update readme

---
 README.md         | 5 ++++-
 make_canonical.py | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 883df19..5e1f6a1 100644
--- a/README.md
+++ b/README.md
@@ -59,7 +59,10 @@ We provide docker environment and setup is as easy as below a few lines.
   cd  docker && bash run_container.sh
   cd /home/catgrasp && bash build.sh
   ```
-  Now the environment is ready to run training or testing.
+  Now the environment is ready to run training or testing. Later you can re-enter the lauched docker environment without re-compilation by:
+  ```
+  docker exec -it catgrasp bash
+  ```
 
 
 # Data
diff --git a/make_canonical.py b/make_canonical.py
index 895f6d8..c2fbd6e 100644
--- a/make_canonical.py
+++ b/make_canonical.py
@@ -84,7 +84,7 @@ def compute_canonical_model():
       if obj_file==other_file:
         continue
       other_cloud = copy.deepcopy(clouds[other_file])
-      other_cloud = (transforms_to_nocs[other_file]@to_homo(other_cloud).T)[:,:3]
+      other_cloud = (transforms_to_nocs[other_file]@to_homo(other_cloud).T).T[:,:3]
       cd = chamfer_distance_between_clouds_mutual(cloud,other_cloud)
       dists.append(cd)
     avg_dist = np.concatenate(dists,axis=0).reshape(-1).mean()