DeepMosaics

Deepmosaics

ภาษาอังกฤษ | 中文
คุณสามารถใช้เพื่อลบโมเสคในรูปภาพและวิดีโอโดยอัตโนมัติหรือเพิ่มโมเสคลงในพวกเขา
โครงการนี้ขึ้นอยู่กับ "การแบ่งส่วนความหมาย" และ "การแปลแบบเป็นภาพ"
ลองใช้ที่เว็บไซต์นี้!

ตัวอย่าง

ต้นทาง	เพิ่มโมเสกอัตโนมัติ	โมเสคทำความสะอาดอัตโนมัติ

• เมื่อเทียบกับ deepcreampy

ภาพกระเบื้องโมเสค	deepcreampy	ของเรา

• การถ่ายโอนสไตล์

ต้นทาง	ถึง Van Gogh	ถึงฤดูหนาว

ตัวอย่างที่น่าสนใจ: Ricardo Milos to Cat

เรียกใช้ deepmosaics

คุณสามารถเรียกใช้ DeepMosaics ผ่านแพ็คเกจไบนารีที่สร้างไว้ล่วงหน้าหรือจากแหล่งที่มา

ลองใช้บนเว็บ

คุณสามารถลองลบโมเสคบน ใบหน้า ได้ที่เว็บไซต์นี้

แพ็คเกจไบนารีที่สร้างไว้ล่วงหน้า

สำหรับ Windows เราได้ทำการทดสอบ GUI เพื่อทดสอบง่าย ๆ
ดาวน์โหลดเวอร์ชันนี้และรุ่นที่ผ่านการฝึกอบรมมาก่อนผ่าน [Google Drive] [百度云, 提取码 1x0a]

• [เอกสารช่วยเหลือ]
  
• วิดีโอสอน => [YouTube] [Bilibili]

ความสนใจ:

• ต้องการ windows_x86_64, windows10 ดีกว่า
  
• โมเดลที่ได้รับการฝึกอบรมมาก่อนที่แตกต่างกันเหมาะสำหรับเอฟเฟกต์ที่แตกต่างกัน [บทนำสู่โมเดลที่ผ่านการฝึกอบรมมาก่อน]
  
• เวลาทำงานขึ้นอยู่กับประสิทธิภาพของคอมพิวเตอร์ (รุ่น GPU มีประสิทธิภาพที่ดีขึ้น แต่ต้องติดตั้ง CUDA)
  
• หากวิดีโอเอาท์พุทไม่สามารถเล่นได้คุณสามารถลองใช้ botplayer
  
• เวอร์ชัน GUI อัปเดตช้ากว่าแหล่งที่มา
  

เรียกใช้จากแหล่งที่มา

ข้อกำหนดเบื้องต้น

• Linux, Mac OS, Windows
• Python 3.6+
• ffmpeg 3.4.6
• Pytorch 1.0+
• CPU หรือ NVIDIA GPU + CUDA CUDNN
  

การพึ่งพาอาศัยกัน

รหัสนี้ขึ้นอยู่กับ OpenCV-Python, Torchvision ผ่านการติดตั้ง PIP

โคลน repo นี้

git clone https://github.com/HypoX64/DeepMosaics.git
cd DeepMosaics

รับรุ่นที่ผ่านการฝึกอบรมมาก่อน

คุณสามารถดาวน์โหลดรุ่น pre_trained และใส่ลงใน './pretrained_models'
[Google Drive] [百度云, 提取码 1x0a]
[บทนำสู่โมเดลที่ผ่านการฝึกอบรมมาก่อน]

ในการเพิ่ม/ลบโมเสคจะต้องมีไฟล์โมเดล mosaic_position.pth ที่ ./pretrained_models/mosaic/mosaic_position.pth

ติดตั้งการพึ่งพา

(ไม่บังคับ) สร้างสภาพแวดล้อมเสมือนจริง

virtualenv mosaic
source mosaic/bin/activate

จากนั้นติดตั้งการอ้างอิง

pip install -r requirements.txt

หากคุณไม่สามารถสร้าง scikit-image ให้เรียกใช้ export CFLAGS='-Wno-implicit-function-declaration แล้วลองสร้างใหม่

ตัวอย่างง่ายๆ

• เพิ่มโมเสค (สื่อเอาท์พุทจะถูกบันทึกไว้ใน './result')
  

python deepmosaic.py --media_path ./imgs/ruoruo.jpg --model_path ./pretrained_models/mosaic/add_face.pth --gpu_id 0

• ทำความสะอาดโมเสค (สื่อเอาท์พุทจะบันทึกใน './result')
  

python deepmosaic.py --media_path ./result/ruoruo_add.jpg --model_path ./pretrained_models/mosaic/clean_face_HD.pth --gpu_id 0

หากคุณเห็นข้อผิดพลาด Please check mosaic_position_model_path! ตรวจสอบว่ามีไฟล์รุ่นชื่อ mosaic_position.pth ที่ ./pretrained_models/mosaic/mosaic_position.pth

พารามิเตอร์เพิ่มเติม

หากคุณต้องการทดสอบรูปภาพหรือวิดีโออื่น ๆ โปรดดูที่ไฟล์นี้
[ตัวเลือก _introduction.md]

การฝึกอบรมด้วยชุดข้อมูลของคุณเอง

หากคุณต้องการฝึกด้วยชุดข้อมูลของคุณเองโปรดดูที่การฝึกอบรม _with_your_own_dataset.md

กิตติกรรมประกาศ

รหัสนี้ยืมมาอย่างหนักจาก [Pytorch-Cyclegan-and-Pix2Pix] [Pytorch-Unet] [PIX2PIXHD] [Bisenet] [DFDNET] [GFRNET_PYTORCH_NEW]