เริ่มต้นใช้งานเริ่มต้นใช้งานได้ฟรี

การโหลดโมเดลที่ผ่านการเทรนมาแล้ว

คุณกำลังสร้างแอปพลิเคชันสำหรับติดป้ายกำกับภาพจากโซเชียลมีเดีย งานนี้ต้องการทั้งความแม่นยำสูงและความเร็ว คุณจะใช้โมเดล ResNet18 ที่ผ่านการเทรนมาแล้วเพื่อทำนายประเภทของภาพ

แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร

Deep Learning สำหรับภาพด้วย PyTorch

ดูคอร์ส

คำแนะนำการฝึกหัด

  • Import resnet18 และ ResNet18_Weights จาก torchvision.models
  • สร้าง model โดยใช้ resnet18() และกำหนดพารามิเตอร์ weights เป็น weights
  • ตั้งค่า model ให้อยู่ในโหมด evaluation
  • กำหนดค่า input transforms และเก็บไว้ในตัวแปร transform

แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ

ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์

# 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 = ____
แก้ไขและรันโค้ด