Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

ฉันจะใช้ Datatset.map ใน Tensorflow เพื่อสร้างชุดข้อมูลของรูปภาพ คู่ป้ายได้อย่างไร


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

อ่านเพิ่มเติม: TensorFlow คืออะไรและ Keras ทำงานร่วมกับ TensorFlow เพื่อสร้าง Neural Networks อย่างไร

เราจะใช้ชุดข้อมูลดอกไม้ซึ่งมีรูปภาพดอกไม้หลายพันดอก ประกอบด้วยไดเร็กทอรีย่อย 5 ไดเร็กทอรี และมีไดเร็กทอรีย่อยหนึ่งไดเร็กทอรีสำหรับทุกคลาส

เรากำลังใช้ Google Colaboratory เพื่อเรียกใช้โค้ดด้านล่าง Google Colab หรือ Colaboratory ช่วยเรียกใช้โค้ด Python บนเบราว์เซอร์และไม่ต้องมีการกำหนดค่าใดๆ และเข้าถึง GPU ได้ฟรี (หน่วยประมวลผลกราฟิก) Colaboratory สร้างขึ้นบน Jupyter Notebook

print("The 'num_parallel_calls' is set so that multiple images are loaded and processed in parallel")
train_ds = train_ds.map(process_path, num_parallel_calls=AUTOTUNE)
val_ds = val_ds.map(process_path, num_parallel_calls=AUTOTUNE)

for image, label in train_ds.take(1):
   print("The shape of image is : ", image.numpy().shape)
   print("The label is : ", label.numpy())

ผลลัพธ์

The 'num_parallel_calls' is set so that multiple images are loaded and processed in parallel
The shape of image is :   (180, 180, 3)
The label is :   0

เครดิตโค้ด:https://www.tensorflow.org/tutorials/load_data/images

คำอธิบาย

  • โหลดและประมวลผลรูปภาพหลายภาพพร้อมกัน
  • เมธอด 'map' ใช้สำหรับสร้างชุดข้อมูลที่มีคู่ (รูปภาพ ป้ายกำกับ)
  • มีการทำซ้ำและขนาดของรูปร่างและป้ายกำกับจะแสดงบนคอนโซล