在屏幕上的窗口中显示PyTorch图像张量

TorchWindow的Python项目详细描述


火炬窗

TorchWindow是一个主要用C编写的pythorchwindow库,可以通过Vulkan Cuda interop直接从GPU内存(GPU和CPU之间没有来回复制)查看屏幕上的PyTorch Cuda张量。在

构建需求

目前TorchWindow只针对64位平台。将来可能会增加32位。在

Windows

Visual Studio
CUDA SDK
Vulkan SDK
CMake公司

Linux

Linux目前不受支持,但将在下一个版本中提供。在

MacOS

MacOS不受支持,但将来可能会支持。在

从源安装

在项目根目录中:

mkdir build
cd build
cmake ..
cmake --build .
cd ..
pip install .

从二进制文件安装

^{pr2}$

使用

要创建窗口:

from torchwindow import Window
window = Window(640, 480, name="Torch Window")

在窗口中显示图像张量

window.draw(image_tensor)

请注意,图像张量必须有3个维度,特别是(宽度、高度、通道)的顺序。 此布局限制将在将来的版本中放宽。在

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
java Jsonify使用Jackson来定义嵌套对象   在Swing中禁用java图形调试   java Selenium Webdriver拖放在Jenkins上不起作用   java我对一个显示器的问题有一个非常不切实际的询问   java增强的“for”循环导致ArrayIndexOutOfBoundsException   ArrayAdapter适用于Java中的安卓编程,字符串数组   linux在Ubuntu上通过PulseAudio播放Java音频文件时出错   java在Spring应用程序中加载内部(类路径)和外部属性文件   java使用Maven连接到mySQL   Java应用程序的设计   websocket在电报api java中与dc的连接   java XMLStreamException,因为xml中的(&N)   java从控制台输出到JTextArea   Java导出文本文件   java实现parseInt方法   java为什么servlet容器会同步对特定资源/servlet的多个请求的访问?   循环中的Java“while”变量   用Java编程一个国际象棋游戏,gameOver布尔不起作用   java如何获得真正的JPanel大小?