# IR-colorization

**Repository Path**: change__zh/IR-colorization

## Basic Information

- **Project Name**: IR-colorization
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-09-02
- **Last Updated**: 2025-09-02

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# IR-colorization


We provide the Pytorch implementation of "Colorization of infrared images based on feature fusion and contrastive learning"
# Getting Started
## Installation
This code was tested with Pytorch 1.6.0, CUDA 10.2, and Python 3.8
## Testing
* Please download the pre-trained model and test set from [here](https://drive.google.com/drive/folders/1vVRhSVFs6yt2R-17dTURf5NeNzAscvkk?usp=sharing), and put model under ./checkpoints/experiment_name/ . put test set under ./datasets/cyclegan/
* test the model 
 ```python test.py --dataroot datasets/cyclegan --model COLOR ```
 # Acknowledge
 Our code is developed based on [CycleGAN](https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix) and [CUT](https://github.com/taesungp/contrastive-unpaired-translation)