show_offloaded_samples.py

import matplotlib.pyplot as plt

def show_image_compression(samples):
    #get encoded samples offloaded to edge
    _, offload_end = device_models['end'].predict(samples)
    #get encoded samples offloaded to cloud
    _, offload_edge = device_models['edge'].predict(offload_end)
    #show sample represenations at each device level
    for i in range(len(samples)):
        print(i)
        for device, img in zip(device_names,[samples[i],offload_end[i],offload_edge[i]]):
            print(device)
            plt.imshow(img[:,:,0])
            plt.show()
        print('\n\n')
        
#show offloaded images for first 10 test samples
show_image_compression(x_test[:10])