Skip to content

Instantly share code, notes, and snippets.

@Akash-Rawat

Akash-Rawat/Predictions_Pedestrian

Created July 13, 2021 06:11
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Akash-Rawat/2f4b98142bc52d35e10c93ac32900b4f to your computer and use it in GitHub Desktop.
Save Akash-Rawat/2f4b98142bc52d35e10c93ac32900b4f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Predictions_Pedestrian
preds = model.predict(val_images)
plt.figure(figsize=(20,20))
for n , i in enumerate(list(np.random.randint(0,len(val_images),36))) :
plt.subplot(6,6,n+1)
plt.imshow(val_images[i])
plt.axis('off')
x =np.argmax(preds[i]) # takes the maximum of of the 6 probabilites.
plt.title((class_names[x]))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment