long (@Longchentong) 在 红外IR和可见光RGB图像对齐的代码 中发帖
红外IR和可见光RGB图像对齐的代码
基于三种变化进行的对齐
fit模式
输入是四个文件夹,RGB图像与标签(yolo格式)文件夹,IR图像与标签(yolo格式)文件夹
输出是配准后的文件
apply模式
输入IR图像文件夹即可
输出配准后的图像文件夹
效果展示
[image]
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
learn_ir_to_rgb_rule.py
在有标签的数据上学习 IR->RGB 的【全局】几何变换规则,并将其保存为 rule.json;
随后可在 test 集(无标签)上直接应用此规则进行配准。
支持三种模型:
- similarity:统一缩放 + 旋转 + 平移(无剪切)
- affine: 含各向异性缩放/剪切 + 平移
- homography:单应性(最灵活)
选择策略:...