下游任务:
采用目标检测、实例分割和深度估计的下游任务来验证图像融合结果质量。
1.目标检测
YOLOv8:https://github.com/ultralytics/ultralytics
可选择的模型类型如下所示,代码示例选择的是YOLOv8n
:
代码-main.py
"""
Author: YidaChen
Time is: 2024/5/11
this Code: 主函数用于模型-目标检测
"""
from ultralytics import YOLO
# Load a model
model = YOLO("yolov8n.yaml")<