search

nv-nguyen / template-pose

[CVPR 2022] Pytorch implementation of "Templates for 3D Object Pose Estimation Revisited: Generalization to New objects and Robustness to Occlusions" paper
https://nv-nguyen.github.io/template-pose/
177 stars 11 forks source link

Visualization of similarity #7

Open liyuan-png opened 1 year ago

liyuan-png commented 1 year ago

Hi @nv-nguyen, Thanks for the interesting paper!! I used test_linemod.py to evaluate the model. However, I didn't find visual results as shown in Figure 5 in the paper. Could you share the code for testing a single image?

Looking forward to your reply and thank you very much!

shqmffl486 commented 1 year ago

I have same question!!!

GilsangJang commented 1 year ago
1 2

In base_network.py, function calculate_similarity_for_search, i can get these pictures with visualizing "sim[i]" matrix and i is the largest summation index of similarity. visualized with cv2 and custom colormap with -1(red), 1(green)

GilsangJang commented 1 year ago

n base_network.py, function

white edges may can be referenced from mask image

11try commented 1 year ago

I have same question!!!