Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
tjulitianyi 7596a39809 | 2 years ago | |
---|---|---|
Spatial-CNN | 2 years ago | |
img | 2 years ago | |
LICENSE | 2 years ago | |
README.md | 2 years ago |
This repository contains my master thesis project called ViDeNN - Deep Blind Video Denoising.
With this pretrained tensorflow model you will be able to denoise videos affected by different types of degradation, such as Additive White Gaussian Noise and videos in Low-Light conditions. The latter has been tested only on one particular camera raw data, so it might not work on different sources. ViDeNN works in blind conditions, it does not require any information over the content of the input noisy video.
IMPORTANT! If you want to denoise data affected by Gaussian noise (AWGN), it has to be clipped between 0 and 255 before denoising it, otherwise you will get strange artifacts in your denoised output.
(noise=15) PSNR | Performance(s / epoch) | |
---|---|---|
Competition baseline(Powered by GPU) | 33.66 | 432 |
Ours result(Powered by Ascend 910A) | 33.69 | 170 |
|-- img ----For README.md
|-- Spatial-CNN ----Code path
|--ckpt ----Automatically generated model storage path
|--logs ----Automatically generated log storage path
|--add_noise_spatialCNN.py ----Dataset preprocessing file
|--dataset_preparation.sh ----Dataset preparation and processing scripts, only run for the first time
|--generate_patches_spatialCNN.py ----Dataset preprocessing file
|--main_spatialCNN.py ----main train code
|--model_spatialCNN.py ----model code
|--train_spatial_cnn.sh ----NPU training startup script
|--utilis.py ----NPU training startup script
|-- LICENSE ----Other file
|-- README.md ----Read me
Note that this code runs based on Huawei Cloud ModelArts and uses Ascend 910A.
$ cd Spatial-CNN
$ bash dataset_preparation.sh
$ bash train_spatial_cnn.sh
python main_spatialCNN.py -h
if you want to set the number of epochs, learning rate, batch size etc.python main_spatialCNN.py --checkpoint_dir=ckpt_awgn --epoch=500
Here, we can see the final performance and accuracy indicators.
The final test results can be obtained by executing the following commands at the terminal:
$ cat test_spatialCNN.log | grep 'Average PSNR' -B 1
Feel free to open an issue if you have any problem, I will do my best to help you.
My E-mail: tjulitianyi@163.com
第四届中国软件开源创新大赛·赛道二:任务挑战赛(模型王者挑战赛):基于华为Ascend 910,利用Tensorflow 1.15.0 实现ViDeNN CV类深盲视频去噪网络,数据集:训练集:WaterLoo Exploration Dataset 测试集:CBSD68,最终精度:PSNR=33.69,性能:173.78 s/epoch
Python Shell
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》