# Indices

**Repository Path**: rolfma/indices

## Basic Information

- **Project Name**: Indices
- **Description**: 本项目是一个计算 CV 领域图像和视频的各种评价指标的工具箱。
- **Primary Language**: Python
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2021-11-17
- **Last Updated**: 2024-01-28

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

## Indices

本项目是一个超分领域辅助处理事务性工作的工具箱。

### 特性支持

- [x] 计算 PSNR 和 SSIM
- [x] 模板匹配
- [x] 寻找适合展示的区域

### 使用方法

#### 单个视频序列

```python
# 导入视频模块
import video

# 设置存放视频序列的路径
sr_path = "./data/001/frvsr"
gt_path = "./data/001/truth"

# 在 Y 通道下，计算四个指标，并存入 result.txt 文件
with open("./result.txt", "a") as f:
    f.write(sr_path+"------"+gt_path+"\n")
    f.write("PSNR:\t"+str(video.get_psnr(sr_path, gt_path, mode="Y"))+"\n")
    f.write("SSIM:\t"+str(video.get_ssim(sr_path, gt_path, mode="Y"))+"\n")
    f.write("WS_PSNR:\t"+str(video.get_ws_psnr(sr_path, gt_path, mode="Y"))+"\n")
    f.write("WS_SSIM:\t"+str(video.get_ws_ssim(sr_path, gt_path, mode="Y"))+"\n")
```

#### 多个视频序列

```python
import video

sr_path_list = ["./data/001/frvsr", "./data/002/frvsr", "./data/003/frvsr", "./data/004/frvsr",
                "./data/001/pfnl", "./data/002/pfnl", "./data/003/pfnl", "./data/004/pfnl"]
gt_path_list = ["./data/001/truth", "./data/002/truth", "./data/003/truth", "./data/004/truth",
                "./data/001/truth", "./data/002/truth", "./data/003/truth", "./data/004/truth"]

for sr_path, gt_path in zip(sr_path_list, gt_path_list):
    print(sr_path+" : "+gt_path)
    with open("./result.txt", "a") as f:
        f.write(sr_path+" : "+gt_path+"\n")
        f.write("PSNR:\t"+str(video.get_psnr(sr_path, gt_path, mode="Y"))+"\n")
        f.write("SSIM:\t"+str(video.get_ssim(sr_path, gt_path, mode="Y"))+"\n")
        f.write("WS_PSNR:\t"+str(video.get_ws_psnr(sr_path, gt_path, mode="Y"))+"\n")
        f.write("WS_SSIM:\t"+str(video.get_ws_ssim(sr_path, gt_path, mode="Y"))+"\n")
```

#### 模板匹配

```python
import image
# 使用 template 匹配 target 图像，确定位置后，根据此位置将 other_dir 目录下的所有图片裁剪
image.template_matching(target_name="target/034.png",
                template_name="template/034temp.png", other_dir="other_dir/9")
```

#### 寻找适合展示的区域

请进入目录 `./find_template `查看方法并使用。