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

การจำแนกประเภทรูปภาพ

ในแบบฝึกหัดนี้ จะใช้ชุดข้อมูล flickr เดิมที่มีรูปภาพ 30,000 ภาพพร้อมคำอธิบายประกอบ โดยครั้งนี้จะนำโมเดลที่ผ่านการ pretrain มาจำแนกประเภทของรูปภาพ

Photo of 2 people, 1 is playing the guitar

รูปภาพตัวอย่าง (image) และโมดูล pipeline (pipeline) ได้ถูกโหลดไว้แล้ว

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

โมเดล Multi-Modal ด้วย Hugging Face

ดูคอร์ส

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

  • โหลด pipeline image-classification โดยใช้โมเดล pretrained google/mobilenet_v2_1.0_224
  • ใช้ pipeline เพื่อทำนายประเภทของรูปภาพตัวอย่าง
  • แสดงผล label แรก (ที่มีความน่าจะเป็นสูงสุด)

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

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

# Load the image-classification pipeline
pipe = ____

# Use the pipeline to predict the class of the sample image
pred = ____

# Print the first (highest probability) label
print("Predicted class:", ____)
แก้ไขและรันโค้ด