Skip to content

Instantly share code, notes, and snippets.

@hzitoun
Created May 18, 2020 16:08
Show Gist options
  • Save hzitoun/2a63d5fa139afad975e528b63c049188 to your computer and use it in GitHub Desktop.
Save hzitoun/2a63d5fa139afad975e528b63c049188 to your computer and use it in GitHub Desktop.
Code for my medium story Serve a Deep Learning Image Classification Model Written in TensorFlow 2 as a REST API and Dockerize it! (with GPU support)
deer = read_image("../data/raw/train/3_deer.png", (32, 32))
car = read_image("../data/raw/train/4_automobile.png", (32, 32))
bird = read_image("../data/raw/train/18_bird.png", (32, 32))
plt.subplot(131)
plt.imshow(deer)
plt.title('Deer')
plt.subplot(132)
plt.imshow(car)
plt.title('Car')
plt.subplot(133)
plt.imshow(bird)
plt.title('Bird')
plt.show()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment