Are you sure you want to delete this access key?
Legend |
---|
DVC Managed File |
Git Managed File |
Metric |
Stage File |
External File |
Legend |
---|
DVC Managed File |
Git Managed File |
Metric |
Stage File |
External File |
Instance segmentation model for detection of the character Baby Yoda, from the Disney TV Show The Mandalorian.
poetry
dvc repro auto-train
The model will be saved in models/model.pth
src/ColabNotebook.ipynb
In a DVC repository run:
dvc import https://dagshub.com/simon/baby-yoda-segmentor models/model.pth
curl -O https://dagshub.com/Simon/baby-yoda-segmentor/raw/master/models/model.pth
from PIL import Image
from torchvision.transforms import ToTensor
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
img = Image.open("image.png")
img_t = ToTensor()(img)
model.eval()
with torch.no_grad():
prediction = model([img_t.to(device)])
Fork the repository
git clone <fork-url>
dvc pull -r origin
dvc remote add --local fork <dagshub-remote-url.dvc>
# Additional commands to set up credentials should appear on you fork homepage
dvc push -r fork
git add .
git commit -m "Changes to dataset"
git push
Press p or to see the previous file or, n or to see the next file
Are you sure you want to delete this access key?
Are you sure you want to delete this access key?
Are you sure you want to delete this access key?
Are you sure you want to delete this access key?