primozgodec / imagecolorization Goto Github PK

View Code? Open in Web Editor NEW

27.0 4.0 14.0 105 KB

Image and video colorizer is package for automatic image and video colorization. Models are allready trained

Python 100.00%

datamining neural-network colorization image-colorization machine-learning artificial-intelligence

imagecolorization's Introduction

Image and video colorizer

Image and video colorizer is package for automatic image and video colorization. Models are already trained.

Instalation

Installation can be done in 5 easy steps

Install all requirements for Tensorflow and tensorflow itself with:
```
pip install tensorflow-gpu
```
if you use GPU device for computation otherwise:
```
pip install tensorflow
```

Create virtual environment

virtualenv -p python3 colorization_venv

Activate virtual environment
```
source colorization_venv/bin/activate
```

Clone Image and video colorization package and move in it

git clone https://github.com/PrimozGodec/ImageColorization.git
cd ImageColorization

Install requirements
```
pip install -r requirements.txt
```
You are done :)

In case you do not have a GPU device in your computer, please install Tensorflow for a CPU. Instructions are at the Tnesorflow website.

Image colorization

For automatic image colorizing follow those steps:

Copy images into /data/image/original directory
Run main.py script from src/image_colorization/ directory.
```
python -m src.image_colorization.main --model <model name>
```
Parameter --method is optional, if not present reg_full_model is default. It can be choose from this list:
- reg_full_model (default)
- reg_full_vgg_model
- reg_part_model
- class_weights_model
- class_wo_weights_model
You can find colored images in /data/image/colorized directory.

on your GPU or CPU specifications. You will see progress bar that show you how far you are with colorization.

Video colorization

For automatic video colorizing follow those steps:

Copy images into /data/video/original directory
Run video_colorizer.py script from src/video_colorization/ directory.
```
python -m src.video_colorization.video_colorizer
```
Video colorizer is always using reg_full_model.
You can find colored videos in /data/video/colorized directory.

Colorization take few hours since there is a lot of images to color in a video and depends on your GPU or CPU specifications and length of a video. You will see progress bar that show you how far you are with colorization.

imagecolorization's People

Contributors

Stargazers

Watchers

Forkers

airacks jonathhhan yashdusing kommalapudi123 kriztoper lydiamargret rajdeepa1 prati01 somyashastri yarensonmez onecanadian watson1101 thorsten65

imagecolorization's Issues

Train script and other details

Hello! I went through your repo and found it very useful. I wanted to inquire if you could also provide the script for training, the model script and indicate what type of data are the pre-trained models trained on. Thanks a lot!

Pre-trained Model

Any plan to upload the pre-trained model ?

How to export model to .pb file for use in ffmpeg filters

Is it possible to export model to use in ffmpeg filters like:
https://ffmpeg.org/ffmpeg-filters.html#toc-dnn_005fprocessing-1

video colorization

Hi.
Thanks in advance.
I tried to run video colorization code and i'm getting the following error.

A video is saved in temp directory but i'm not able to run it.

video

Hello, I'm sorry to bother you, but I encountered the following problems when I was dealing with the video:

Moreover, the MP4 file generated cannot be opened. May I ask why?
Thank you for your reply.

Issue

thanks. having some issues

Unable to run src.image_colorization.main, complains that src can't be found
and for running the main.py for images, what should be indicated in

HTTP Error : Error 404

While I am running the code, I got an error.

data/weights/imp9-full.h5
Downloading trained model
Traceback (most recent call last):
File "C:\Users\Dhanya\Anaconda3\envs\tensorflow\lib\runpy.py", line 193, in _run_module_as_main
"main", mod_spec)
File "C:\Users\Dhanya\Anaconda3\envs\tensorflow\lib\runpy.py", line 85, in _run_code
exec(code, run_globals)
File "C:\Users\Dhanya\Downloads\ImageColorization-master\ImageColorization-master\src\image_colorization\main.py", line 21, in
model.load_weights(get_weights(imported_model.weights))
File "C:\Users\Dhanya\Downloads\ImageColorization-master\ImageColorization-master\src\utils\image_utils.py", line 92, in get_weights
reporthook=show_progress)
File "C:\Users\Dhanya\Anaconda3\envs\tensorflow\lib\urllib\request.py", line 248, in urlretrieve
with contextlib.closing(urlopen(url, data)) as fp:
File "C:\Users\Dhanya\Anaconda3\envs\tensorflow\lib\urllib\request.py", line 223, in urlopen
return opener.open(url, data, timeout)
File "C:\Users\Dhanya\Anaconda3\envs\tensorflow\lib\urllib\request.py", line 532, in open
response = meth(req, response)
File "C:\Users\Dhanya\Anaconda3\envs\tensorflow\lib\urllib\request.py", line 642, in http_response
'http', request, response, code, msg, hdrs)
File "C:\Users\Dhanya\Anaconda3\envs\tensorflow\lib\urllib\request.py", line 570, in error
return self._call_chain(*args)
File "C:\Users\Dhanya\Anaconda3\envs\tensorflow\lib\urllib\request.py", line 504, in _call_chain
result = func(*args)
File "C:\Users\Dhanya\Anaconda3\envs\tensorflow\lib\urllib\request.py", line 650, in http_error_default
raise HTTPError(req.full_url, code, msg, hdrs, fp)
urllib.error.HTTPError: HTTP Error 404: Not Found

Also the url provided in the image_utils.py file is not available.
https://github.com/PrimozGodec/ImageColorization/releases/download/v0.0.1

invalid argument

size of the video file in temp and the original video is different thats why invalid argument error occurs what to do to solve this issue