轻量级CUDA张量函数,侧重于图像数据增强。
augp的Python项目详细描述
augpy是一个轻量级库,具有最小的依赖性 提供CUDA张量函数,重点放在图像数据上 增强。 它使用dlpack 将张量导出到其他库的标准,例如 Pythorch,零拷贝。在
建筑
确保安装了以下组件。在
- 用C++ 14支持(例如GCC 5) 编译程序
- Cuda 10或更高
- Cmake 3.13或更高
- setuptools>;=44.0.0
- 车轮>;=0.34.0
- pybind11>;=2.4.3
- numpy>;=1.15.0
现在只需像平常一样运行setup.py来构建轮子并用pip安装。在
python setup.py bdist_wheel
使用
托多
变更日志
- 在制品
- 项目
标签: