pytorch-image-quality-param-ctrl

PyTorch implementation of

• DeepBIQ(https://arxiv.org/pdf/1602.05531.pdf)
• use DeepBIQ and ppo(https://github.com/ikostrikov/pytorch-a2c-ppo-acktr) to search usb camera image quality param(brightness, contrast, saturation, sharpness)

Contributions

Contributions are very welcome. If you know how to make this code better, don't hesitate to send a pull request.

Requirements

• Ubuntu 14.04
• usb camera (must support 640 * 480 resolution)
• python3.6
• PyTorch
• GPU
• LIVE In the Wild Image Quality Challenge Database (ChallengeDB_release, it must locate in: pytorch-image-quality-param-ctrl/deepbiq/ChallengeDB_release)
• if you want ppo to search good image quality param, you should add more sample to ChallengeDB_release, for example: very dark, very bright, very saturated images and so on to train DeepBIQ model

warning

you should save your usb camera default quality param

v4l2-ctl --get-ctrl brightness
v4l2-ctl --get-ctrl contrast
v4l2-ctl --get-ctrl saturation
v4l2-ctl --get-ctrl sharpness

maybe you have to install v4l2-ctl

sudo apt-get install v4l2-ctl

Training DeepBIQ

cd deepbiq/
sh train.sh

use ppo to Search Image Quality Param

cd ppo/
sh search.sh

if occurred error like：set brightness=106 failed:Input/output error ; you have to power up again usb camera