File size: 765 Bytes
627e429 |
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 |
import numpy as np
import matplotlib.pyplot as plt
def simple_cmfd_decoder(busterNetModel, rgb):
"""A simple BusterNet CMFD decoder"""
# 1. expand an image to a single sample batch
single_sample_batch = np.expand_dims(rgb, axis=0)
# 2. perform busterNet CMFD
pred = busterNetModel.predict(single_sample_batch)[0]
return pred
def visualize_result(rgb, gt, pred, figsize=(12, 4), title=None):
"""Visualize raw input, ground truth, and BusterNet result"""
fig = plt.figure(figsize=figsize)
plt.subplot(1, 3, 1)
plt.imshow(rgb)
plt.title("input image")
plt.subplot(1, 3, 2)
plt.title("ground truth")
plt.imshow(gt)
plt.subplot(1, 3, 3)
plt.imshow(pred)
plt.title("busterNet pred")
return fig
|