isl-org / DPT

Dense Prediction Transformers
MIT License
2.01k stars 258 forks source link

RGB Depth estimation #7

Closed ChiscoD closed 3 years ago

ChiscoD commented 3 years ago

In the paper, depths are showed as red when is near, and black when in far, how can I replicate the papers result? (color).

ranftlr commented 3 years ago

The visualization uses the "inferno" colormap from matplotlib:

import matplotlib.pyplot as plt
plt.imshow(pred, cmap=plt.cm.inferno)
ChiscoD commented 3 years ago

Thanks