1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning cho Ảnh với PyTorch
Connected

Bài tập

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.

Hướng dẫn

100 XP
  • 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.