Geotessera library for OCaml
patrick.sirref.org
1module Gt = Geotessera
2
3let belfast =
4 Gt.Bbox.v ~min_lat:54.48 ~min_lon:(-6.11) ~max_lat:54.66 ~max_lon:(-5.78)
5
6let pp_result fmt ((crs, transform), (point : Gt.point), emb) =
7 Fmt.pf fmt "Embedding at (%.2f, %.2f) with CRS:%i: %a\ntransform: [%a]"
8 point.lat point.lon crs Nx.pp_shape (Nx.shape emb) Nx.pp transform
9
10let () =
11 Mirage_crypto_rng_unix.use_default ();
12 Eio_main.run @@ fun env ->
13 let embeddings = Gt.fetch env ~year:2024 belfast in
14 Eio.traceln "Dequantizing embeddings...";
15 let dequantize =
16 List.map (fun (crs, p, _lm, e) -> (crs, p, Gt.scale e)) embeddings
17 in
18 Eio.traceln "Embeddings: %a"
19 Fmt.(list ~sep:(Fmt.any "\n") pp_result)
20 dequantize