Skip to content

Qengineering/ncnn-Colorization_Raspberry-Pi-4

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Colorization on a Raspberry Pi

output image

Colorization with the ncnn framework.

License

Paper: https://arxiv.org/pdf/1603.08511.pdf

Special made for a bare Raspberry Pi 4, see Q-engineering deep learning examples


Dependencies.

To run the application, you have to:

  • A raspberry Pi 4 with a 32 or 64-bit operating system. It can be the Raspberry 64-bit OS, or Ubuntu 18.04 / 20.04. Install 64-bit OS
  • The Tencent ncnn framework installed. Install ncnn
  • OpenCV 64 bit installed. Install OpenCV 4.5
  • Code::Blocks installed. ($ sudo apt-get install codeblocks)

Installing the app.

To extract and run the network in Code::Blocks
$ mkdir MyDir
$ cd MyDir
$ wget https://github.com/Qengineering/ncnn-Colorization_Raspberry-Pi-4/archive/refs/heads/main.zip
$ unzip -j master.zip
Remove master.zip, LICENSE and README.md as they are no longer needed.
$ rm master.zip
$ rm LICENSE
$ rm README.md

Your MyDir folder must now look like this:
15.jpg
17.jpg
Colorization.cpb
main.cpp
Sig17Slice.h
siggraph17_color_sim.param


Download the model weights

Before you can run the app, you have to download the model weights. The file is too large (125 MB) to be stored on GitHub (limited to 100 MB). Download location of siggraph17_color_sim.bin at GDrive.


Running the app.

To run the application load the project file Colorization.cbp in Code::Blocks.
The app takes the b/w input file and the output file name as command line arguments.
You may find the hint at Hands-On handy.
Do note the application takes about 10 Sec to process a single picture.


Thanks.

A more than special thanks to magicse, who adapted the ncnn framework for this colorization app.


More info.

Colorful Image Colorization magicse
Colorful Image Colorization richzhang/colorization
Colorful Image Colorization Project Page by Richard Zhang, Phillip Isola, Alexei A. Efros.


output image

paypal