Mục đích: mô hình demo phân loại một số giống chó/mèo phổ biến tại Việt Nam (corgi, poodle, shiba, aln, scottish).
Trạng thái: minh họa giáo dục/nghiên cứu; không dùng cho chẩn đoán hay quyết định y tế.
1from PIL import Image
2import torch
3from torchvision import transforms
4# model = torch.jit.load("model.pt").eval() # khi có trọng số
5preprocess = transforms.Compose([transforms.Resize((224,224)), transforms.ToTensor()])
6img = Image.open("pet.jpg").convert("RGB")
7x = preprocess(img).unsqueeze(0)
8# logits = model(x); probs = torch.softmax(logits, dim=1)
