GitHub - Jianbo-Lab/LBIE

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
CoSaL		CoSaL
bw_images		bw_images
flower		flower
models		models
new_output		new_output
README.txt		README.txt
cells.py		cells.py
colornet.py		colornet.py
datapipe.py		datapipe.py
extract_cap.py		extract_cap.py
generate_image.py		generate_image.py
main.py		main.py
model2.py		model2.py
requirements.txt		requirements.txt
temp.py		temp.py
texts.txt		texts.txt
utils.py		utils.py

Repository files navigation

Install packages in requirements.txt

1. Create a folder 'input_dir' for storing original images.
2. Create a text_file 'text_file' where each line is: image_name:sentence description.  
3. Run the following in Shell:
python generate_image.py --input_dir 'input_dir' --text_file 'text_file' --output_dir 'output_dir'
4. The BW images and their colorized versions can be found in the folder 'output_dir'

Examples can be found in the zip file.