pet-breed-small-v1

pet-breed-small-v1

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ế.

Cách dùng (minh họa)

from PIL import Image
import torch
from torchvision import transforms
# model = torch.jit.load("model.pt").eval()  # khi có trọng số
preprocess = transforms.Compose([transforms.Resize((224,224)), transforms.ToTensor()])
img = Image.open("pet.jpg").convert("RGB")
x = preprocess(img).unsqueeze(0)
# logits = model(x); probs = torch.softmax(logits, dim=1)