Wolowolo commited on
Commit
3aafd1b
·
verified ·
1 Parent(s): bc65635

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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