Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
shiboxin
c230da2915
|
3 years ago | |
|---|---|---|
| .META-INF | 3 years ago | |
| .idea | 3 years ago | |
| data | 3 years ago | |
| model | 3 years ago | |
| networks | 3 years ago | |
| utils | 3 years ago | |
| PanoPack.py | 3 years ago | |
| README.md | 3 years ago | |
| config.yaml | 3 years ago | |
README.md
全景图像反射消除
Panoramic Image Reflection Removal
作者
施柏鑫
项目简介
1. 功能
利用全景图像的360°视角获取反射场景，从而辅助反射消除。
2. 性能
- 资源占用：单组数据测试时，显存占用约2400MB，模型参数量约为100M。
3. 评估指标
- 由于实验数据没有背景的ground truth，因此无法计算指标，但可根据视觉质量进行分析。
4. 使用数据集
- 无开源数据集，需自行捕获。
- 项目demo中包含两组测试数据，存放于data文件夹下，每组数据序号应以XXXXX的格式命名，如00001。
- 每组数据内有两张图像，分别为带反射图像和反射场景，分别命名为m.jpg和rs.jpg。
运行环境与依赖
代码运行的环境与依赖。如下所示：
| 类别 | 名称 | 版本 |
|---|---|---|
| os | ubuntu | 16.04 |
| 深度学习框架 | pytorch | 1.6.0 |
| 依赖包 | opencv | 3.4.2 |
输入与输出
代码的输入与输出。如下所示：
| 名称 | 说明 |
|---|---|
| 输入 | 带反射混合图像和反射场景图像，大小为300X300（宽x高）（可在config.yaml文件内设置）。 |
| 输出 | 消除反射后的图像，大小为300X300（宽x高）（可在config.yaml文件内设置）。 |
运行方式
- 下载预训练模型 [link]（提取码：pcl1），并存放在model文件夹下.
- 在terminal下运行以下命令。
cd project_dir
python ./panopack.py