小编Y0s*_*tsu的帖子

PyTorch 模型到 C++

我已经训练了检测算法并保存了我最好的模型。现在我想将我的模型(预训练)转换为 C++ 并在我的应用程序中使用它。我想知道将 pyTorch 模型转换为 C++ 的可能方法是什么?

谢谢!

c++ python-3.x pytorch onnx

7
推荐指数
1
解决办法
4679
查看次数

移动第一个元素以 Numpy 数组结尾

我有一个 NumPy 数组,我想将第一个元素移动到最后。我知道如何通过转换为列表并执行插入操作来做到这一点,但我想避免这种情况并以更好和优化的方式进行。

假设这是我的数组,我想将 0 移到最后:

x = [[ 0 388.13 19.346 412.38 39.594]
    [ 0 388.15 8.5168 416 38.972] 
    [ 0 223.14 156.21 317.91 193.46]]
Run Code Online (Sandbox Code Playgroud)

我知道我可以通过转换为列表以这种方式做到这一点:

x = [[ 0 388.13 19.346 412.38 39.594]
    [ 0 388.15 8.5168 416 38.972] 
    [ 0 223.14 156.21 317.91 193.46]]
Run Code Online (Sandbox Code Playgroud)

有没有办法在不将其转换为列表(使用 numpy)的情况下做到这一点?而且,我正在做的方式效率不高,即使将代码转换为列表,也有什么可能的方法可以使我的代码在速度方面高效?

预期输出:

x = [[388.13 19.346 412.38 39.594 0]
    [388.15 8.5168 416 38.972 0] 
    [223.14 156.21 317.91 193.46 0]]
Run Code Online (Sandbox Code Playgroud)

python numpy python-3.x

2
推荐指数
2
解决办法
460
查看次数

标签 统计

python-3.x ×2

c++ ×1

numpy ×1

onnx ×1

python ×1

pytorch ×1