# SMFuse

**Repository Path**: lulululala/SMFuse

## Basic Information

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

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2023-10-16
- **Last Updated**: 2023-10-16

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# SMFuse
The code of “SMFuse: Multi-focus Image Fusion via Self-supervised Mask-optimization”
#### To train :<br>
Run "CUDA_VISIBLE_DEVICES=0 python main.py" to train the network.

#### To test :<br>
Run "CUDA_VISIBLE_DEVICES=0 python test_one_image.py" to test the network.

说明：
mask文件夹是基于用户自己的训练集获得的，目录中的mask是作者基于作者自己的训练集所获得。
mask的获取方式是作者模型中的方法，运行method_get_mask中main.py