rekaxua / demosaic_project

Removing Pixelated Mosaic Censorship using ESRGAN and green_mask_project

demosaic_project

Removing Pixelated Mosaic Censorship using ESRGAN (Enhanced SRGAN) and green_mask_project

ESRGAN model should be downloaded here: Twittman's 4x_FatalPixels_340000_G.pth, and placed into models folder!

To have a decent performance you need a CUDA+cudnn(NVIDIA) or ROCm(AMD) compatible GPU with installed drivers! Otherweis use "--cpu" key, but beware it can be VERY slow.

HOW TO USE:

1. Place your images inside decensor_input
2. Run "python demosaic_project_ESRGAN.py" or ("python demosaic_project_ESRGAN.py --cpu" for cpu version)
3. Take your decensored images from decensor_output
4. ???
5. Profit

Don't forget to install python3 and all the requirements with command "pip3 install -r requirements.txt" in cmd or bash

TODO:

• GUI
• Config file
• Better detection algorithm

Inspiration from DeepCreamPy and hent-AI
Credits to Twittman for making trained model Fatal Pixels.

Sample image by hks(@timbougami)

demosaic_project's Issues

Program refuses to function

the command python demosaic_project_ESRGAN.py --cpu
does not run the program in cpu mode, because there seems to be something wrong with the code.

no matter what I do, I get this error: raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled

URL problems

There seems to be a problem with the URL
Twittman's 4x_FatalPixels_340000_G.pth: https://de-next.owncube.com/index.php/s/mDGmi7NgdyyQRXL/download?path=%2F&files=4x_FatalPixels_340000_G.pth

Line missing

I receive this error:
"Traceback (most recent call last):
File "E:\demosaic_project\demosaic_project_ESRGAN.py", line 28, in
import architecture as arch
ModuleNotFoundError: No module named 'architecture'"

Is there a line missing that indicates the location of architecture.py?

Help with configurating\error?

D:\demosaic_project-1.0>pip3 install -r requirements.txt
Collecting glob3
  Using cached glob3-0.0.1.tar.gz (669 bytes)
ERROR: Could not find a version that satisfies the requirement torch==1.2.0 (from -r requirements.txt (line 2)) (from versions: 0.1.2, 0.1.2.post1, 0.1.2.post2)
ERROR: No matching distribution found for torch==1.2.0 (from -r requirements.txt (line 2))



D:\demosaic_project-1.0>python demosaic_project_ESRGAN.py
Traceback (most recent call last):
  File "demosaic_project_ESRGAN.py", line 1, in <module>
    import cv2
ModuleNotFoundError: No module named 'cv2'

I download demosaic_project-1.0.rar, extracted and downloaded 4x_FatalPixels_340000_G.pth (placed it in model folder)
Have installed last python 3.8.2

Glob3 can't be downloaded. Can I upload it?

glob3下载不到了，可以给一个吗？