mejora de la imagen de vídeo
Red de reconstrucción de súper resolución
1. Introducción
Para animarse a sí mismo a estudiar mucho, Bibo escribe lo que ha aprendido recientemente en un blog /(ㄒoㄒ)/~~
2. DASR
2.1 Papeles y código
Si quieres saber más sobre la idea de esta red, te recomiendo que leas el enlace del blogger de abajo, habla mejor que yo, así que no me explicaré demasiado.
Interpretación del documento: https://blog.csdn.net/weixin_43972154/article/details/119327182
En cuanto a aprender el código, está bien seguir los pasos de operación del autor paso a paso. Para aprender una nueva red, lo más importante La cosa es reproducir el código.
Enlace de código: https://github.com/The-Learning-And-Vision-Atelier-LAVA/DASR
3. Reproducción de código
3.1 Preparar el entorno
Dado que el entorno de pytorch utilizado por el autor es relativamente antiguo, siempre habrá problemas inimaginables al utilizar una versión superior de pytorch Para ahorrar tiempo, le sugiero que utilice un entorno virtual. Es muy simple de operar, solo tres pasos:
- Cree un entorno virtual
conda create -n pytorch1.1 python=3.6
y encuentre opciones de selección en el caminoy
- Activar el entorno virtual creado
activate pytorch1.1
- Instale los paquetes requeridos en el entorno virtual activado
pip install -i https://pypi.douban.com/simple -r requirement.txt
(recuerde poner los paquetes requeridos en un archivo txt)
3.2 Preparación de datos
Descarga estos dos datos y colócalos en una carpeta, para evitar menos errores, llamamos a la carpeta comoDF2K
- Preparar
DIV2K
DIV2K
Enlace de descarga del conjunto de datos: https://data.vision.ee.ethz.ch/cvl/DIV2K/
Hay muchos conjuntos de datos en este enlace. No supe qué conjuntos de datos descargar en ese momento y casi los descargué todos. Tenga en cuenta que solo necesita Descargar estos dos conjuntos de datos En cuanto a por qué, puede leer la interpretación del documento que proporcioné anteriormente.
- Preparar
Flickr2K
Flickr2K
Enlace de descarga del conjunto de datos: https://cv.snu.ac.kr/research/EDSR/Flickr2K.tar
- Combinar datos
Después de descargar los datos, colóquelos en la carpeta DF2K
, que se ve así.
Una red cuidadosa puede preguntar por qué solo tengo 900 imágenes, porque mi computadora no tiene suficiente memoria de video y solo uso el DIV2K
conjunto de datos para entrenamiento.
3.3 Empezar a entrenar
main.sh
El siguiente paso es comenzar a entrenar. El código de entrenamiento se puede ejecutar directamente si usa un servidor . Si no tiene condiciones, puede usar el siguiente comando:
python main.py --dir_data=./dataset/ --model=blindsr --scale=4 --blur_type=aniso_gaussian --noise=25.0 --lambda_min=0.2 --lambda_max=4.0
El proceso de entrenamiento es probablemente así
Tenga en cuenta los siguientes problemas:
3.4 Datos de prueba
- El documento original preparó cuatro conjuntos de datos de prueba, hablemos sobre el [enlace de descarga] de los datos. (https://github.com/XPixelGroup/BasicSR/blob/a19aac61b277f64be050cef7fe578a121d944a0e/docs/Datasets.md)
- Solo usé
Set14
el conjunto de datos cuando probé, usé estas dos carpetas.
- Coloque
Set14
lasHR
imágenes yLR
las imágenes enbenchmark
la carpeta, y el formato de archivo de almacenamiento de datos es así.
3.5 Iniciar la prueba
Lo siguiente es para iniciar la prueba, el comando de prueba es el siguiente:
python test.py --test_only --dir_data=./dataset/ --data_test=Set14 --model=blindsr --scale=4 --resume=114 --blur_type=aniso_gaussian --noise=10.0 --theta=0.0 --lambda_1=0.2 --lambda_2=4.0 --save_results=True
Las siguientes preguntas:
Además, también puedes usar los pesos oficiales para una prueba rápida.
3.6 Resultados de la prueba
Cabe mencionar aquí que los resultados del proceso de entrenamiento se guardan experiment
en el archivo, y el modo que utilice estará en la carpeta.
Aquí está lo siguiente, los resultados de mi entrenamiento, porque solo entrené 114epoch, entonces el efecto no es bueno, mostrémoslo con dificultad.
4. Mi Código
El siguiente es el enlace a mi código. Los estudiantes perezosos pueden descargar mi código directamente. Ya he descargado todos los datos, por lo que no necesito molestarme en descargar los conjuntos de datos uno por uno, solo utilícelos directamente.
Enlace: https://pan.baidu.com/s/1f_Gq-pvMyfth-FebiYpx7A
Código de extracción: fasf
: compartir del super miembro V5 de Baidu Netdisk