Update new_approach/spa_ensemble.py

new_approach/spa_ensemble.py

@@ -69,8 +69,15 @@ class FeatureExtractor:
 
     @staticmethod
     def extract_frequency_features(img):
-        img_np = np.array(img); gray = cv2.cvtColor(img_np, cv2.COLOR_RGB2GRAY); gray_small = cv2.resize(gray, (64, 64)); dct_coeffs = dct(dct(gray_small.T, norm='ortho').T, norm='ortho'); features = {}
-        for i, v in enumerate(dct_coeffs.flatten()[:10]): features[f'freq_dct_{i}'] = float(v); return features
+        img_np = np.array(img)
+        gray = cv2.cvtColor(img_np, cv2.COLOR_RGB2GRAY)
+        gray_small = cv2.resize(gray, (64, 64))
+        dct_coeffs = dct(dct(gray_small.T, norm='ortho').T, norm='ortho')
+        features = {}
+        # FIX: Loop must finish before returning!
+        for i, v in enumerate(dct_coeffs.flatten()[:10]): 
+            features[f'freq_dct_{i}'] = float(v)
+        return features
 
     @staticmethod
     def extract_statistical_features(img):