update

CFA.py

@@ -46,12 +46,23 @@ def GetMap(image_channel):
     return e.ravel()
 
 def getFourier(prob):
+    #imFft = np.fft.fftshift(np.fft.fft2(prob))
+    #imFft = np.abs(imFft)
+    #if np.max(imFft) > 0:
+    #    imFft = (imFft / np.max(imFft) * 255)
+    #imFft = imFft.astype(np.uint8)
+    #imFft = (imFft > (0.5 * 255)).astype(np.uint8)
+    # Compute the Fourier Transform and shift zero frequency to the center
     imFft = np.fft.fftshift(np.fft.fft2(prob))
+    
+    # Take the magnitude (absolute value)
     imFft = np.abs(imFft)
-    if np.max(imFft) > 0:
-        imFft = (imFft / np.max(imFft) * 255)
-    imFft = imFft.astype(np.uint8)
-    imFft = (imFft > (0.5 * 255)).astype(np.uint8)
+    
+    # Convert to 8-bit unsigned integer (similar to uint8 in MATLAB)
+    imFft = np.uint8(imFft)/255
+
+    # Binarize the image with a threshold of 0.5
+    imFft = (imFft > 0.5).astype(np.uint8)
     return imFft
 
 # --- New Gradio Interaction Functions ---
@@ -110,13 +121,13 @@ def analyze_region(original_image: np.ndarray, box_coords: tuple):
     prob_map = prob_flat.reshape(prob_map_shape)
     
     print("3. Computing Fourier transform...")
-    fft_result = getFourier(patch)#prob_map)
+    fft_result = getFourier(prob_map)
 
     # Plotting
     fig, axs = plt.subplots(1, 3, figsize=(12, 4))
     axs[0].imshow(patch); axs[0].set_title("Selected 128x128 Patch"); axs[0].axis("off")
     axs[1].imshow(prob_map, cmap='gray'); axs[1].set_title("Probability Map"); axs[1].axis("off")
-    axs[2].imshow(fft_result, cmap='gray'); axs[2].set_title("Fourier Transform"); axs[2].axis("off")
+    axs[2].imshow(np.abs(fft_result), cmap='gray'); axs[2].set_title("Fourier Transform"); axs[2].axis("off")
     plt.tight_layout()
     
     print(f"4. Analysis complete in {time.time() - start_time:.2f} seconds.")