หน้าแรก>การเขียนโปรแกรมที่เกี่ยวข้อง>ซอร์สโค้ดอื่น ๆ
ดาวน์โหลด

การฉายภาพ: การอนุมานภาพวาดรูปร่างจากภาพ

ทีเซอร์

repo นี้มีรหัสการฝึกอบรมและการทดสอบสำหรับเครื่องกำเนิดภาพร่างของเรา นอกจากนี้เรายังมี [โมเดลที่ผ่านการฝึกอบรมมาแล้ว]

สำหรับรายละเอียดทางเทคนิคและชุดข้อมูลโปรดดู [กระดาษ] และ [หน้าโครงการ]

การตั้งค่า

ตอนนี้รหัสได้รับการอัปเดตเพื่อใช้ Pytorch 0.4 และทำงานบน Windows, Mac และ Linux สำหรับรุ่นที่ล้าสมัยด้วย Pytorch 0.3 (Linux เท่านั้น) โปรดตรวจสอบสาขา Pytorch-0.3-Obsolete

ผู้ใช้ Windows ควรค้นหาไฟล์ .cmd ที่เกี่ยวข้องแทนไฟล์ .sh ที่กล่าวถึงด้านล่าง

การติดตั้งแบบบรรทัดเดียว (พร้อมสภาพแวดล้อม conda)

conda env create -f environment.yml

จากนั้นเปิดใช้งานสภาพแวดล้อม (ร่าง) และคุณพร้อมที่จะไป!

ดูที่นี่สำหรับข้อมูลเพิ่มเติมเกี่ยวกับสภาพแวดล้อม conda

การติดตั้งด้วยตนเอง

ดู environment.yml สำหรับรายการของการพึ่งพา

ใช้โมเดลที่ผ่านการฝึกอบรมมาก่อน

  • ดาวน์โหลดรุ่นที่ผ่านการฝึกอบรมมาก่อน
  • แก้ไขเส้นทางใน scripts/test_pretrained.sh
  • จาก ไดเรกทอรีราก ของ repo เรียกใช้ scripts/test_pretrained.sh

รองรับโฟลเดอร์ของรูปภาพเป็นอินพุต

ฝึกอบรมและทดสอบในชุดข้อมูลการวาดรูปทรงของเรา

  • ดาวน์โหลดภาพและภาพร่างที่แสดงผลจากหน้าโครงการ
  • เปิดเครื่องซิปและจัดระเบียบให้เป็นโครงสร้างต่อไปนี้:

โครงสร้างไฟล์

  • แก้ไขเส้นทางใน scripts/train.sh และ scripts/test.sh
  • จาก ไดเรกทอรีราก ของ repo เรียกใช้ scripts/train.sh เพื่อฝึกอบรมโมเดล
  • จาก ไดเรกทอรีรูท ของ repo เรียกใช้ scripts/test.sh เพื่อทดสอบในชุด Val หรือชุดทดสอบ (ระบุโดยการตั้งค่าสถานะเฟส)

การอ้างอิง

หากคุณใช้รหัสหรือข้อมูลสำหรับการวิจัยของคุณโปรดอ้างอิงกระดาษ: 

 @article{LIPS2019,
  title={Photo-Sketching: Inferring Contour Drawings from Images},
  author={Li, Mengtian and Lin, Zhe and Mv ech, Radom'ir and and Yumer, Ersin and Ramanan, Deva},
  journal={WACV},
  year={2019}
}

การรับทราบ

รหัสนี้ใช้ PIX2PIX เวอร์ชันเก่า

ขยาย
ข้อมูลเพิ่มเติม
แอปที่เกี่ยวข้อง
แนะนำสำหรับคุณ
ข้อมูลที่เกี่ยวข้อง ทั้งหมด