contrib/papers/UColor

• Ucolor
• • Requirements
  • Train
  • Test
  • Acknowledgement

Ucolor

This is the implementation of Ucolor using MindSpore

Requirements

1. python 3.7.5
2. minspore-gpu 1.3.0
3. cuda 10.1

You'd better install the mindspore-gpu follow the official instruction.

You can create a conda environment to run our code like this:

conda env create -f ./mindspore1.3.yaml

Train

1. Deownload the pretrain vgg for mindspore.
2. Put your data in the data folder.
3. You are ready to train your own UColor Model!

python ./train.py

Use python ./train.py --help for more details.

Test

Put your data in the data folder.

python ./test.py

Use python ./test.py --help for more details.

Acknowledgement

Thanks to the minspore team who maintain the latest version of the framework,
and also the model zoo.