English|中文

Black and White Image Colorization

This application runs on the Atlas 200 DK to automatically color black and white images.

The applications in the current version branch adapt to DDK&RunTime 1.32.0.0 and later.

Prerequisites

Before deploying this sample, ensure that:

• Mind Studio has been installed.
• The Atlas 200 DK developer board has been connected to Mind Studio, the cross compiler has been installed, the SD card has been prepared, and basic information has been configured.

Deployment

You can use either of the following methods:

1. Quick deployment: visit https://github.com/Atlas200dk/faster-deploy.

   NOTE:

   • The quick deployment script can be used to deploy multiple samples rapidly. Select colorization.
   • The quick deployment script automatically completes code download, model conversion, and environment variable configuration. To learn about the detailed deployment process, go to 2. Common deployment.

2. Common deployment: visit https://github.com/Atlas200dk/sample-README/tree/master/sample-colorization.

   NOTE:

   • In this deployment mode, you need to manually download code, convert models, and configure environment variables.

Build

1. Open the project.

   Go to the directory that stores the decompressed installation package as the Mind Studio installation user in CLI mode, for example, $HOME/MindStudio-ubuntu/bin. Run the following command to start Mind Studio:

   ./MindStudio.sh

   Open the sample_colorization project, as shown in Figure 1.

   Figure 1 Opening the sample_colorization project

   

2. Configure project information in the src/param_configure.conf file.

   Figure 2 Configuration file path
   

   The default configurations of the configuration file are as follows:

   remote_host=192.168.1.2
   

   remote_host: IP address of the Atlas 200 DK developer board

   NOTE:

   • Do not use double quotation marks ("") during parameter settings.
   • Modify the default configurations as required.

3. Run the deploy.sh script to adjust configuration parameters and download and compile the third-party library. Open the Terminal window of Mind Studio. By default, the home directory of the code is used. Run the deploy.sh script in the background to deploy the environment, as shown in Figure 3.

   Figure 3 Running the deploy.sh script
   

   NOTE:

   • During the first deployment, if no third-party library is used, the system automatically downloads and builds the third-party library, which may take a long time. The third-party library can be directly used for the subsequent build.
   • During deployment, select the IP address of the host that communicates with the developer board. Generally, the IP address is the IP address configured for the virtual NIC. If the IP address is in the same network segment as the IP address of the developer board, it is automatically selected for deployment. If they are not in the same network segment, you need to manually type the IP address of the host that communicates with the Atlas DK to complete the deployment.

4. Start building. Open Mind Studio and choose Build > Build > Build-Configuration from the main menu. The build and run folders are generated in the directory.

   NOTICE:
   When you build a project for the first time, Build > Build is unavailable. You need to choose Build > Edit Build Configuration to set parameters before the build.

5. Upload the images to be colored to any directory of the HwHiAiUser user on the host side.

Run

1. On the toolbar of Mind Studio, click Run and choose Run > Run 'sample-colorization'. As shown in Figure 4, the executable application is running on the developer board.

   Figure 4 Running application
   

   You can ignore the error information reported during the execution because Mind Studio cannot transfer parameters for an executable application. In the preceding steps, the executable application and dependent library files are deployed to the developer board. You need to log in to the developer board in SSH mode and manually execute the files in the corresponding directory. For details, see the following steps.

2. Log in to the host side as the HwHiAiUser user in SSH mode on Ubuntu Server where Mind Studio is located.

   ssh HwHiAiUser@host_ip

   For the Atlas 200 DK, the default value of host_ip is 192.168.1.2 (USB connection mode) or 192.168.0.2 (NIC connection mode).

3. Go to the path of the executable file of the black and white image colorization application.

   Command example:

   cd ~/HIAI_PROJECTS/workspace_mind_studio/sample-colorization_xxxxx/out

   NOTE:

   • In this path, _xxxxx _in sample-colorization_xxxxx is a combination of letters and digits generated randomly each time the application is built.

4. Run the application.

   Run the run_colorization.py script to save the images which are generated by inference to the specified path.

   Command example:

   python3 run_colorization.py -i ~/example.jpg -o ./out/

   • -i: path of the input image. The value can be a directory, indicating that all images in the current directory are used as input. The value can also be used to specify an image.
   • -o: path for storing the colored images.