1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
|
#!/usr/bin/env python
import matplotlib.pyplot as plt
import imgviz
def depth2rgb():
data = imgviz.data.arc2017()
depthviz = imgviz.depth2rgb(data["depth"], min_value=0.3, max_value=1)
# -------------------------------------------------------------------------
plt.figure(dpi=200)
plt.subplot(121)
plt.title("rgb")
plt.imshow(data["rgb"])
plt.axis("off")
plt.subplot(122)
plt.title("depth (colorized)")
plt.imshow(depthviz)
plt.axis("off")
img = imgviz.io.pyplot_to_numpy()
plt.close()
return img
if __name__ == "__main__":
from base import run_example
run_example(depth2rgb)
|