About test results - Githubissues

xuefeng-cvr / BS-Net

Pytorch implementation of "[PR2021] Boundary-induced and scene-aggregated network for monocular depth prediction"

41 stars 4 forks source link

Closed Kammagod closed 3 years ago

Kammagod commented 3 years ago

hi,how do I get depth maps of test results. Thanks!

xuefeng-cvr commented 3 years ago

You can save the variable 'pred' at Line 60 in 'test.py' by the following code:

plt.matshow(pred.squeeze().cpu().numpy()) plt.savefig("fig.png"))

Kammagod commented 3 years ago

You can save the variable 'pred' at Line 60 in 'test.py' by the following code:

plt.matshow(pred.squeeze().cpu().numpy()) plt.savefig("fig.png"))

thanks! I will try