ดาวน์โหลด SRGAN - ดาวน์โหลดซอร์สโค้ด SRGAN

SRGAN

หลาม

1.0.0

ดาวน์โหลด

srgan

การใช้งาน Pytorch ของ SRGAN โดยใช้ CVPR 2017 Paper Photo-Realistic Single Image Super-Resolution โดยใช้เครือข่ายฝ่ายตรงข้าม

ความต้องการ

อนาคอนดา
pytorch

 conda install pytorch torchvision -c pytorch

opencv

 conda install opencv

ชุดข้อมูล

ฝึกอบรมชุดข้อมูล VAL

ชุดข้อมูลรถไฟและ VAL ถูกสุ่มตัวอย่างจาก VOC2012 ชุดข้อมูลรถไฟมีภาพ 16700 ภาพและชุดข้อมูล VAL มี 425 ภาพ ดาวน์โหลดชุดข้อมูลจากที่นี่ (รหัสการเข้าถึง: 5TZP) จากนั้นแยกออกเป็นไดเรกทอรี data

ชุดข้อมูลภาพทดสอบ

ทดสอบชุดข้อมูลวิดีโอ

ชุดข้อมูลวิดีโอทดสอบเป็นสามตัวอย่าง ดาวน์โหลดชุดข้อมูลวิดีโอจากที่นี่ (รหัสการเข้าถึง: 956D)

การใช้งาน

รถไฟ

 python train.py

optional arguments:
--crop_size                   training images crop size [default value is 88]
--upscale_factor              super resolution upscale factor [default value is 4](choices:[2, 4, 8])
--num_epochs                  train epoch number [default value is 100]

ภาพเอาต์พุต Val Super Resolution อยู่ในไดเรกทอรี training_results

ชุดข้อมูลมาตรฐานทดสอบ

 python test_benchmark.py

optional arguments:
--upscale_factor              super resolution upscale factor [default value is 4]
--model_name                  generator model epoch name [default value is netG_epoch_4_100.pth]

ภาพเอาต์พุต Super Resolution อยู่ในไดเรกทอรี benchmark_results

ทดสอบภาพเดียว

 python test_image.py

optional arguments:
--upscale_factor              super resolution upscale factor [default value is 4]
--test_mode                   using GPU or CPU [default value is 'GPU'](choices:['GPU', 'CPU'])
--image_name                  test low resolution image name
--model_name                  generator model epoch name [default value is netG_epoch_4_100.pth]

ภาพเอาต์พุต Super Resolution อยู่ในไดเรกทอรีเดียวกัน

ทดสอบวิดีโอเดี่ยว

 python test_video.py

optional arguments:
--upscale_factor              super resolution upscale factor [default value is 4]
--video_name                  test low resolution video name
--model_name                  generator model epoch name [default value is netG_epoch_4_100.pth]

วิดีโอความละเอียด Super Resolution และวิดีโอเปรียบเทียบอยู่ในไดเรกทอรีเดียวกัน