Update app.py
Browse files
app.py
CHANGED
@@ -257,7 +257,7 @@ def FSFM3C_image_detection(image):
|
|
257 |
cam_path = os.path.join(CAM_SAVE_PATH, str(len(os.listdir(CAM_SAVE_PATH))))
|
258 |
os.makedirs(cam_path, exist_ok=True)
|
259 |
os.makedirs(os.path.join(cam_path, '0'), exist_ok=True)
|
260 |
-
output_path = os.path.join("output_heatmap.png")
|
261 |
cv2.imwrite(output_path, visualization)
|
262 |
return image_results, output_path, probabilities[max_prob_index]
|
263 |
|
|
|
257 |
cam_path = os.path.join(CAM_SAVE_PATH, str(len(os.listdir(CAM_SAVE_PATH))))
|
258 |
os.makedirs(cam_path, exist_ok=True)
|
259 |
os.makedirs(os.path.join(cam_path, '0'), exist_ok=True)
|
260 |
+
output_path = os.path.join(cam_path, "output_heatmap.png")
|
261 |
cv2.imwrite(output_path, visualization)
|
262 |
return image_results, output_path, probabilities[max_prob_index]
|
263 |
|