当前位置: 首页 > news >正文

2D水平目标检测数据增强——旋转任意指定角度

这段代码主要实现了以下功能:

  1. 环境设置与导入库:设置环境变量(针对Apple MPS设备),导入必要的库(如os, numpy, torch, matplotlib, PIL, OpenCV等)。
  2. 设备选择:根据可用性选择运行设备(CUDA、MPS或CPU),并进行一些设备相关的配置(如使用bfloat16、TF32等)。
  3. 辅助函数定义
    • get_rect:通过阈值分割和轮廓检测,从二值图像中提取包围所有轮廓的最小矩形。
    • rotate_image:将图像旋转指定角度,并返回旋转后的图像以及旋转矩阵等信息。
    • rotate_mask:类似于rotate_image,但专用于处理掩码(单通道)。
    • read_yolo_labels:读取YOLO格式的标签文件,返回标签列表(每个标签包含类别和归一化坐标)。
    • read_yolo_labels_for_infer:从YOLO标签文件中读取标签,并转换为图像上的实际坐标(用于推理)。
    • visualize_yolo_labels:在图像上绘制YOLO标签(矩形框和类别)。
    • mkdir:创建目录(如果不存在)。
    • convert_to_yolo
http://www.dtcms.com/a/341714.html

相关文章:

  • RK3568 Linux驱动学习——设备树下 LED 驱动
  • Redisson最新版本(3.50.0左右)启动时提示Netty的某些类找不到
  • PowerShell脚本检查业务健康状态
  • 解决Docker 无法连接到官方镜像仓库
  • Lecture 6 Kernels, Triton 课程笔记
  • JVM基础知识总结
  • Docker 核心技术:Linux Cgroups
  • GDB 的多线程调试
  • 针对具有下垂控制光伏逆变器的主动配电网络的多目标分层协调电压/无功控制方法的复现
  • 音频读写速度优化 音频格式
  • Transformer内容详解(通透版)
  • pip install -e中e 参数解释
  • 八辊矫平机·第三篇
  • 卸载win10/win11系统里导致磁盘故障的补丁
  • 广东省省考备考(第八十二天8.20)——资料分析、数量、言语(强化训练)
  • 【蒸蒸日上】军八武将篇——标1
  • 8 webUI中-Controlnet(控制与约束)的应用分类与使用方法
  • 【语法】markdown非常用场景
  • Netty HashedWheelTimer设计原理:从时间轮算法到源码实现
  • 跨平台 RTSP/RTMP 播放器工程化实践:低延迟与高稳定性的挑战与突破
  • 【数据分享】东北大鼠疫传播与死亡空间数据
  • Vue透传 Attributes(详细解析)2
  • 恶补DSP:2.F28335的定时器系统
  • 买返商城网站源码多平台购物返现搭建图解源码二开
  • 万象生鲜配送系统 2025 年 8 月 15 日更新日志
  • 八月月报丨MaxKB在教育及教学科研领域的应用进展
  • Hadoop学习
  • 达梦数据库-实时主备集群部署详解(附图文)手工搭建一主一备数据守护集群DW
  • HyDE vs HyPE:AI检索界的‘假想敌’革命,如何让RAG系统从‘找资料’变成‘懂你心’?”
  • Firefox 142 引入 CRLite 用于私有证书撤销