csv_export.py

# csv_export.py

import csv
import os
import logging

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

def save_features_to_csv(image_id, features, bbox, label, csv_file):
    """
    Save the extracted features into a CSV file.
    The CSV header is: Image_ID, Sharpness, Edge_Density, SIFT_Keypoint_Count,
    Object_Count, Tampered_Ratio, Forgery_BBox, Label.
    """
    header = ["Image_ID", "Sharpness", "Edge_Density", "SIFT_Keypoint_Count",
              "Object_Count", "Tampered_Ratio", "Forgery_BBox", "Label"]
    file_exists = os.path.exists(csv_file)
    
    try:
        with open(csv_file, mode="a", newline="") as file:
            writer = csv.writer(file)
            if not file_exists:
                writer.writerow(header)
            writer.writerow([
                image_id,
                features.get('sharpness', 0),
                features.get('edge_density', 0),
                features.get('sift_count', 0),
                features.get('object_count', 0),
                features.get('tampered_ratio', 0),
                str(bbox),  # Save bbox as string
                label
            ])
        logger.info("Saved features for %s", image_id)
    except Exception as e:
        logger.error("Error writing to CSV: %s", e)