Experimenting with if we can use tf.keras to output both predictions and heatmap, and the resulting model can be
saved for serving.
python grad_cam.py
would wrap a resnet to output both prediction and heatmap and output a SavedModel
. test_saved_model
would send a request to SavedModel
and verify the output indeed contains heatmap.
dsimic / tf-keras-gradcam Goto Github PK
View Code? Open in Web Editor NEWThis project forked from jackhaha363/tf-keras-gradcam
gradCAM implementaion using tf.keras for deployment