Function main

tests/test_sam2_pe_cache_size.cpp:19–116

#include <cstring>

int main(int argc, char** argv) {
    if (argc < 3) {
        fprintf(stderr, "Usage: %s <sam2_model.ggml> <image.jpg>\n", argv[0]);
        return 1;
    }

    sam3_params p;
    p.model_path = argv[1];
    p.n_threads = 4;
    p.use_gpu = false;

    auto model = sam3_load_model(p);
    if (!model) { fprintf(stderr, "FAIL: cannot load model\n"); return 1; }

    // Must be a SAM2 model
    auto mt = sam3_get_model_type(*model);
    if (mt != SAM3_MODEL_SAM2) {
        fprintf(stderr, "SKIP: not a SAM2 model (type=%d)\n", mt);
        return 0;
    }

    auto state = sam3_create_state(*model, p);
    if (!state) { fprintf(stderr, "FAIL: cannot create state\n"); return 1; }

    // Load and encode image
    auto image = sam3_load_image(argv[2]);
    if (image.data.empty()) { fprintf(stderr, "FAIL: cannot load image\n"); return 1; }
    fprintf(stderr, "Image: %dx%d\n", image.width, image.height);

    if (!sam3_encode_image(*state, *model, image)) {
        fprintf(stderr, "FAIL: image encoding failed\n");
        return 1;
    }

    // Run PVS with a point at the center of the image
    float cx = image.width / 2.0f;
    float cy = image.height / 2.0f;
    fprintf(stderr, "Point: (%.0f, %.0f)\n", cx, cy);

    sam3_pvs_params pvs;
    pvs.pos_points.push_back({cx, cy});
    pvs.multimask = false;  // single mask — same as GUI default

    auto result = sam3_segment_pvs(*state, *model, pvs);

    // ── Test 1: Must produce exactly 1 detection ──
    if (result.detections.empty()) {
        fprintf(stderr, "FAIL: no detections returned\n");
        return 1;
    }
    fprintf(stderr, "Detections: %zu\n", result.detections.size());

    auto& det = result.detections[0];
    fprintf(stderr, "Mask 0: iou=%.4f, mask=%dx%d\n",
            det.iou_score, det.mask.width, det.mask.height);

    // ── Test 2: IoU score must be reasonable (> 0.1) ──
    // A garbage PE grid produces IoU near 0 or random values