Views
No views yet
1pip install torch torchvision pillow
2pip install lightly-train1lightly-train embed \
2 data="images/" \
3 checkpoint="/cxr-dino-model/model.pt" \
4 out="embeddings.pth" \
5 format="torch"1import lightly_train
2
3lightly_train.embed(
4 data="images/",
5 checkpoint="cxr-dino-model/model.pt",
6 out="embeddings.pth",
7 format="torch"
8)
9
10print("Saved embeddings to embeddings.pth")1import torch
2
3data = torch.load("embeddings.pth")
4embeddings = data["embeddings"]
5paths = data["image_paths"]
6
7print(embeddings.shape)
8print(paths[:5])