Function main

tests/test_debug_encoder.cpp:64–242

}

int main(int argc, char** argv) {
    if (argc < 4) {
        fprintf(stderr, "Usage: %s <model.ggml> <ref_dir> <output_dir>\n", argv[0]);
        fprintf(stderr, "\nref_dir: directory with Python reference tensors\n");
        fprintf(stderr, "output_dir: where to dump C++ tensors\n");
        return 1;
    }

    const std::string model_path = argv[1];
    const std::string ref_dir = argv[2];
    const std::string out_dir = argv[3];
    ensure_dir(out_dir);

    // ═══ Load model ═══
    fprintf(stderr, "\n═══ Loading model ═══\n");
    sam3_params params;
    params.model_path = model_path;
    params.use_gpu = false;  // CPU for determinism
    params.n_threads = 4;

    auto model = sam3_load_model(params);
    if (!model) {
        fprintf(stderr, "Failed to load model\n");
        return 1;
    }

    auto state = sam3_create_state(*model, params);
    if (!state) {
        fprintf(stderr, "Failed to create state\n");
        return 1;
    }

    // ═══ Stage 1: Load Python-preprocessed image ═══
    fprintf(stderr, "\n═══ Stage 1: Loading Python-preprocessed image ═══\n");
    auto ref_img = load_ref_f32(ref_dir + "/preprocessed");
    if (ref_img.data.empty()) {
        fprintf(stderr, "Failed to load %s/preprocessed.bin\n", ref_dir.c_str());
        return 1;
    }
    fprintf(stderr, "  Loaded preprocessed image: shape=[");
    for (size_t i = 0; i < ref_img.shape.size(); ++i) {
        if (i > 0) fprintf(stderr, ",");
        fprintf(stderr, "%d", ref_img.shape[i]);
    }
    fprintf(stderr, "] (%d elements)\n", ref_img.numel());

    const int img_size = 1008;
    const int expected_elems = 3 * img_size * img_size;
    if (ref_img.numel() != expected_elems) {
        fprintf(stderr, "  ERROR: expected %d elements, got %d\n", expected_elems, ref_img.numel());
        return 1;
    }

    // ═══ Encode image using Python-preprocessed data ═══
    fprintf(stderr, "\n═══ Encoding image (from Python preprocessed data) ═══\n");
    bool ok = sam3_encode_image_from_preprocessed(*state, *model, ref_img.data.data(), img_size);
    if (!ok) {
        fprintf(stderr, "sam3_encode_image_from_preprocessed failed!\n");