计算机视觉深度学习图像处理【免费下载链接】kornia Geometric Computer Vision Library for Spatial AI项目地址https://gitcode.com/GitHub_Trending/ko/kornia点击查看免费下载导读Kornia 是一个基于 PyTorch 的可微分计算机视觉库覆盖几何变换、滤波、增强、特征与三维视觉等方向。本文围绕官方社区贡献文档 docs/source/community/contribute.rst完整讲解如何参与这个开源项目——从报告 bug、提出使用问题到用 Pixi 搭建开发环境、编写测试、遵守编码规范、提交 Pull Request 的完整链路。读完本文你将掌握一套可直接落地的贡献流程并理解 Kornia 在测试矩阵、变更日志changelog与 API 表面管理上的工程约定。一、贡献的总体原则每一条路径都算数Kornia 的贡献文档开篇即表明态度We develop Kornia for ourselves and the computer vision community任何形式的帮助都受欢迎。贡献不只是写代码以下每条路径都是有效的贡献方式报告 bug附带最小可复现示例在 GitHub Discussions 或 Discord 上提问与解答问题提交代码或文档改进通过 Pull Request赞助项目通过 Open Collective 与 GitHub Sponsors。仓库根目录的 CONTRIBUTING.md 进一步阐述了社会契约social contract维护者会尽力审查 PR、修复 bug但不承诺审查每一个 PR 或保证速度。这意味着贡献者需要做好沟通预期把耐心和认真放在首位。1.1 务实指南动手前先做的事开始之前先搜索已有的 issue 与 PR避免重复劳动对于新公开 API、行为变更或大块工作提前开 issue 或 discussion对齐方向——对话只代表对齐不承诺一定会被合并修复 bug 时尽量附上没有该修复就会失败的测试数值类改动最好给出参考实现、论文或数学不变量的来源先查找kornia中是否已有可复用的工具函数再决定是否新增实现保持改动聚焦、贴近周边代码风格公开行为变化时同步更新文档PR 静默 15 天会被标记为 stale但这只是队列整理信号不会被关闭评论或推送即可自动清除维护者也可加pinned使其豁免清扫。二、报告 Bug让维护者能一次复现官方文档建议在 GitHub issue 追踪器上新建 issue并强调最有价值的报告包含最小可复现示例以及 bug 模板要求的环境信息。环境行通常包括Kornia、PyTorch、Python 版本操作系统OS类型设备后端CPU / CUDA / MPS / TPUGPU 型号如果相关。发布前请预览并脱敏任何隐私信息。仓库的 AGENTS.md 给出了一条隐私意识良好的环境采集命令与 bug 模板保持一致python -W ignore -c import kornia, platform, sys, torch; gputorch.cuda.get_device_name(0) if torch.cuda.is_available() else none; backendtorch.version.cuda or getattr(torch.version, hip, None); print(fkornia {kornia.__version__} | torch {torch.__version__} | python {sys.version.split()[0]} | OS {platform.platform()} | CUDA/ROCm {backend} (available{torch.cuda.is_available()}, GPU{gpu}) | MPS{torch.backends.mps.is_available()})该命令只打印版本与硬件信息不包含用户名、路径、环境变量或令牌可安全分享。诊断问题的通用方法同样来自 AGENTS.md从最小复现与最窄的相关测试开始不要在没有与基线版本安全对比之前就断言故障是本来就存在的只索要有助于诊断的环境细节绝不要索要敏感信息。此外很多bug其实是约定不匹配。提问前请先查阅 FAQ 与 约定页面例如张量布局是BCHW还是BHWC、坐标系约定等。三、提问与讨论问题该放哪里使用问题与想法请走社区渠道不要把 QA 放进 GitHub issuesGitHub Discussions讨论区#kornia标签PyTorch 论坛Discord 聊天。对应的文档链接分别是文档中的 faqs.rst 与 conventions.rst。保持公开讨论issue、discussion、PR 评论区方便他人检索与学习。四、贡献代码开发环境搭建Pixi4.1 环境管理为什么用 PixiKornia 使用 Pixi 中声明了defaultPython 3.11、py312、py313、py314、cuda等环境。依赖定义在 pyproject.toml 中Pixi 任务底层调用uvPython 测试运行在仓库.venv或版本化的.venv-py*项目环境中。安装 Pixi三选一# Linux / macOS curl -fsSL https://pixi.sh/install.sh | bash # Windows (PowerShell) irm https://pixi.sh/install.ps1 | iex # 或通过 conda/mamba conda install -c conda-forge pixi4.2 标准开发流程Fork 仓库并克隆自己的 fork添加上游远程git clone gitgithub.com:你的 GitHub 用户名/kornia.git cd kornia git remote add upstream https://github.com/kornia/kornia.git新建分支不要在main上直接开发git checkout upstream/main -b feat/foo_feature # 或 git checkout upstream/main -b fix/bar_bug安装开发环境pixi install pixi run -e default install # 特定 Python 版本自动使用隔离的 .venv-py312 / .venv-py313 pixi install -e py312 pixi run -e py312 install pixi install -e py313 pixi run -e py313 install # CUDA 开发会重装 PyTorch使用 CUDA 12.1 pixi run -e cuda install4.3 Pixi 任务速查表pixi.toml 中定义了完整的任务清单日常开发常用命令如下类别命令说明安装pixi run -e default install安装开发依赖安装pixi run -e default install-docs安装开发 文档依赖测试pixi run test全量测试由KORNIA_TEST_*环境变量配置测试pixi run test-f32/test-f64float32 / float64 测试测试pixi run test-slow慢测试KORNIA_TEST_RUNSLOWtrue测试pixi run test-module tests/path聚焦某个测试路径/文件测试pixi run test-halfCPU float16/bfloat16 测试测试pixi run -e cuda test-cudaCUDA 测试测试pixi run -e cuda test-cuda-halfCUDA half 精度测试每测试子进程隔离质量pixi run lint仅运行 Ruff 钩子质量pixi run pre-commit-all运行全部 pre-commit 检查质量pixi run pre-commit-install安装 pre-commit Git 钩子质量pixi run typecheck用ty做类型检查文档pixi run -e default build-docs构建文档Sphinx-W严格模式工具pixi run doctest运行 docstring 示例跳过需下载权重的工具pixi run clean清理 Python 缓存文件4.4 测试环境变量与 pytest 参数测试可用环境变量配置pixi.toml中test任务的说明# 设备cpu, cuda, mps, tpu export KORNIA_TEST_DEVICEcuda # dtypefloat32, float64, float16, bfloat16 export KORNIA_TEST_DTYPEfloat32 # 是否运行慢测试 export KORNIA_TEST_RUNSLOWtrue pixi run test也可以直接用 pytest 参数test-module会把参数透传给 pytestpixi run test-module tests/geometry --devicecpu --dtypefloat32,float64 pixi run test-module tests/TEST_TO_RUN.py::TestClass::test_method pixi run test-module tests/path/ -k expressiondtype 选项bfloat16、float16、float32、float64、alldevice 选项cpu、cuda、tpu、mps、all编译测试设置KORNIA_TEST_OPTIMIZERinductor会启用 dynamo/compile 测试未设置时默认剔除doctest 权重pixi run doctest-weights或KORNIA_DOCTEST_DOWNLOAD1会真实运行需要下载模型权重的 docstring 示例冷缓存约需 838 MB五、编码规范提交前必须遵守的规则5.1 小步提交与注释提交小而逻辑完整的改动写清晰的提交信息避免超大文件遵循 PEP8行宽120 字符使用 f-stringPEP 498注释用英文解释为什么而非重复代码在做什么二进制运算符后换行W504在特定场景可接受。5.2 类型注解Python 3.11优先直接用torch.Tensor、torch.nn.Module做类型注解向后兼容时可从kornia.core导入非 JIT 模块必须加from __future__ import annotations函数输入输出始终要有类型注解用pixi run typecheck基于ty验证from __future__ import annotations import torch def homography_warp( patch_src: torch.Tensor, dst_homo_src: torch.Tensor, dsize: tuple[int, int], mode: str bilinear, padding_mode: str zeros ) - torch.Tensor: ...5.3 依赖纪律避免新增运行时依赖确有必要时说明现有依赖为何不够用并权衡安装、构建与平台成本AGENTS.md 中同样强调。六、测试规范BaseTester 与测试矩阵6.1 测试文件放哪里测试放在 tests/ 下可复用的测试支撑代码放在 testing/不是tests/。核心测试基类是 testing/base.py 中的BaseTester它封装了assert_close带 dtype 特定容差、gradcheck与 dynamo 相关工具。6.2 标准测试骨架CONTRIBUTING.md 给出了推荐的测试类结构from testing.base import BaseTester class TestMyFunction(BaseTester): # 用 self.assert_close(...) 比较实际与期望张量 def test_smoke(self, device, dtype): # 用不同参数组合调用函数确认至少能运行 pass def test_exception(self, device, dtype): # 测试函数可能抛出的异常 # with pytest.raises(Error) as errinfo: # your_function(触发错误的参数) # assert 错误信息 in str(errinfo) pass def test_cardinality(self, device, dtype): # 测试不同参数下输出形状是否符合预期 pass def test_feature_foo(self, device, dtype): # 基础功能测试 pass def test_gradcheck(self, device): # 梯度检查使用 self.gradcheck(...) pass def test_dynamo(self, device, dtype, torch_optimizer): # 使用 dynamo 优化器测试 # inputs (...) # op your_function # op_optimized torch_optimizer(op) # self.assert_close(op(inputs), op_optimized(inputs)) pass6.3 设备与 dtype 覆盖用 pytest 参数化覆盖不同设备、dtype 与 batch sizeimport pytest pytest.mark.parametrize(batch_size, [1, 2, 5]) def test_smoke(batch_size, device, dtype): x torch.rand(batch_size, 2, 3, devicedevice, dtypedtype) assert x.shape (batch_size, 2, 3)6.4 已知失败与精度细节AGENTS.md 要点半精度与 MPS 已知失败通过清单管理如 testing/known_failure_xfails/mps_float32.txt修复后删除对应清单行MPS 不支持 float64 gradcheck且 autocast 可能改变有效 dtypeTF32 矩阵乘默认关闭--tf32开启cuDNN 卷积仍使用 PyTorch 的 TF32 默认值torch.clamp不能作为可移植的梯度保护在 torch 2.5.1/2.9.1 上clamp边界处梯度会原样通过而在 2.14.0 上返回 0因此求导表达式中应避免把 clamp 边界当作奇异点保护保留注入的device/dtypefixture不要隐式创建 CPU 或默认 dtype 张量。更完整的机制含 guarded candidate recording、fresh-process replay见 TESTING.md。七、变更日志Changelog与文档更新7.1 Changelog 片段机制Kornia 用 Towncrier 管理变更日志详见 changelog.d/README.md。每个有用户可见变化的 PR 在 changelog.d/ 下新增一个 Markdown 片段文件而不是直接改CHANGELOG.md从而避免多个 PR 在同一文件冲突。命名规则PR号.类型.md例如1234.fixed.md用 PR 号不是 issue 号类型与含义类型对应标题用途addedAdded新特性与面向用户的文档fixedFixed对既有行为的修正breakingBreaking不兼容变更必须同时描述旧行为与新行为若 PR 号未知先用唯一名称如fix-camera-shape.fixed.md合并前改名为 PR 号内部重构、纯测试与仅 CI 的改动可省略片段需在 PR 描述中说明并请维护者打no-changelog标签预览未发布的变更说明pixi run changelog-preview不修改任何文件只有发布类 PR 才允许编辑CHANGELOG.md发布流程用pixi run changelog-build --version X.Y.Z组装。7.2 文档更新文档源码在 docs/source/ 下改动后运行pixi run -e default build-docs构建用浏览器预览docs/build/html/index.html每个新的公开类或函数都要登记到对应的docs/source/*.rstAPI 页面否则不会出现在渲染后的 API 参考中新增特征检测器/描述子时还要更新 docs/generate_examples.py 中的responses列表参考现有 DISK、ALIKED、XFeat 示例并保持(B, 3, H, W)图像形状。7.3 API 表面API surface保护tests/api_surface.json 记录了稳定核心模块的公开名称tests/test_api_surface.py 会拒绝在未同步更新清单的情况下删除公开 API可用tests.test_api_surface.regenerate()重新生成清单被移除的 API 需同时登记到 tests/api_surface_removals.json例如{kornia.geometry.boxes: [Boxes]}删除公开名称需走弃用窗口与发布说明流程。八、Pull Request 全流程8.1 填写 PR 模板每个 PR 都要按 .github/pull_request_template.md 填写足够上下文模板要求四部分这个 PR 做了什么、关联的 issue 或 discussion、你检查了什么列出运行过的测试与结果、AI 如何帮助了你。$ pixi run test-module tests/...对 bug 修复能复现 bug 的回归测试最有价值数值/几何类改动要说明期望值来源参考实现、论文或数学不变量新公开 API 或行为变更建议先有 issue 讨论。8.2 PR 检查与合并要求代码改动运行 CPU 测试矩阵、dynamo/compile 测试与ty类型检查pre-commit 检查格式、lint、license 头、拼写与仓库文件卫生文档改动运行文档构建每个非 draft 的 PR 打开或被标记 ready 时自动获得 Copilot 审查并需要一名维护者批准PR 采用squash-merge合并静默过久的 PR 可能被关闭但可以随时重新打开检查失败时请修复或解释原因。8.3 如何挑选第一个贡献文档修复、新测试和标记为good first issue的 issue 是最友好的切入点help wanted标签的 issue 适合寻找功能实现起点。九、性能、基准与 ONNX进阶贡献须知9.1 性能实践优先原地操作tensor.add_(other)优于tensor tensor.add(other)用张量视图与切片代替复制用 PyTorch 向量化操作代替 Python 循环优化前先 profiletorch.profiler或cProfile关注大张量的内存效率性能关键路径可考虑torch.jit.script或torch.compile。9.2 基准测试性能相关改动遵循 benchmarks/README.md 的方法论契约与新增基准清单使用 benchmarks/common.py 中的辅助函数基准应调用公开 kornia API而不是内嵌替代实现性能改动需在基线与改动分支上跑同一基准对比。9.3 ONNX 专项ONNX 在 Kornia 中是完整子系统而非导出附庸改动导出行为前先检查 kornia/onnx/、kornia/core/ 中的 mixin 与既有 ONNX 测试开发依赖可选的onnx、onnxruntime、onnxscript包同时影响图构建/导出与运行时行为的改动需要两种测试。十、AI 协作与项目文化Kornia 对 AI 辅助持开放态度CONTRIBUTING.md 明确写道You may use AI to do the work — so do we但要求认真对待产出像检查自己写的代码一样检查 AI 输出并在 PR 中简要说明 AI 帮了什么、你随后改了哪些、自己核对了什么——不需要百分比或归属类别。AGENTS.md 则是给编码 Agent 的日常操作手册涵盖项目结构kornia/geometry/、kornia/filters/、kornia/augmentation/、kornia/feature/ 与 kornia/models/、kornia/onnx/、kornia/core/、导入顺序注意事项kornia/__init__.py中filters与geometry先于依赖它们的模块加载改动导出时需检查循环导入以及可编辑安装下的分支对比陷阱比较脚本应放在 worktree 根目录以 stdin 或-c方式运行并打印kornia.__file__确认导入的是目标树。十一、许可证与赞助通过贡献你同意以 Apache License 2.0 授权你的贡献见 LICENSE。如果写代码不是你的强项也可以考虑通过 Open Collective 或 GitHub Sponsors 赞助项目维护者同时欢迎 CUDA 服务器捐赠用于测试。详见 sponsor 页面。小结贡献清单速查提交 PR 前逐项核对已搜索既有 issue/PR改动聚焦单一主题环境已就绪pixi installpixi run -e default install已通过pixi run lint、pixi run pre-commit-all、pixi run typecheck已运行相关测试pixi run test-module tests/pathdtype/device 覆盖合理新公开 API 已添加类型注解、docstring 与 API 文档登记用户可见变化已添加changelog.d/PR.type.md片段PR 描述按模板填写包含检查结果与 AI 使用说明。赞分享计算机视觉深度学习图像处理【免费下载链接】kornia Geometric Computer Vision Library for Spatial AI项目地址https://gitcode.com/GitHub_Trending/ko/kornia点击查看免费下载相关推荐免费RAW处理神器darktable如何用AI技术让照片编辑效率提升300%免费RAW处理神器darktable如何用AI技术让照片编辑效率提升300% 你是否还在为繁琐的RAW照片处理流程而烦恼作为摄影师处理大量RAW文件需要花图像处理桌面应用VeighNa Alpha模块完整指南快速搭出一个机器学习多因子策略VeighNa Alpha模块完整指南快速搭出一个机器学习多因子策略 vn.py的VeighNa Alpha模块是机器学习多因子策略的本地研究工具从K线数据金融科技人工智能打造可交互tooltipjQuery PowerTip鼠标悬停交互功能详解打造可交互tooltipjQuery PowerTip鼠标悬停交互功能详解 想要为你的网站添加 专业级悬浮提示框 吗jQuery PowerTip是一个功能UI库/组件上一篇终极指南Google Breakpad Microdump轻量级崩溃报告技术解析与最佳实践下一篇Nuklear预编译头配置大型项目的编译加速创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考