Tải mô hình đã được huấn luyện sẵn

Bạn đang xây dựng một ứng dụng gán nhãn ảnh từ mạng xã hội. Nhiệm vụ này đòi hỏi độ chính xác cao và tốc độ nhanh. Bạn sẽ dùng mô hình ResNet18 đã huấn luyện sẵn để suy luận các lớp ảnh.

Bài tập này là một phần của khóa học

Deep Learning cho Ảnh với PyTorch

Xem khóa học

Hướng dẫn bài tập

Import resnet18 và ResNet18_Weights từ torchvision.models.
Khởi tạo model bằng resnet18(), thiết lập tham số weights là weights.
Đặt model sang chế độ đánh giá.
Khởi tạo các phép biến đổi đầu vào và gán cho transform.

Bài tập tương tác thực hành trực tiếp

Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.

# Import resnet18 model
from torchvision.models import ____

# Initialize model with default weights
weights = ResNet18_Weights.DEFAULT
model = ____

# Set model to evaluation mode
model.____

# Initialize the transforms
transform = ____

Chỉnh sửa và Chạy Mã