Update gradcam_xception.py
Browse files- gradcam_xception.py +1 -0
gradcam_xception.py
CHANGED
|
@@ -207,6 +207,7 @@ def save_comparison(image, cam, overlay, face_box=None):
|
|
| 207 |
plt.close()
|
| 208 |
buf.seek(0)
|
| 209 |
return Image.open(buf)
|
|
|
|
| 210 |
#load xception model
|
| 211 |
def load_xception_model(model_repo="drg31/xception", model_filename="final_xception_model.pth", num_classes=2):
|
| 212 |
try:
|
|
|
|
| 207 |
plt.close()
|
| 208 |
buf.seek(0)
|
| 209 |
return Image.open(buf)
|
| 210 |
+
|
| 211 |
#load xception model
|
| 212 |
def load_xception_model(model_repo="drg31/xception", model_filename="final_xception_model.pth", num_classes=2):
|
| 213 |
try:
|