标签: pytorch nvidia-jetson siamese-network jetson-xavier
我正在使用pytorch在Jetson Xavier NX上运行SiamMask(https://github.com/foolwood/SiamMask)。
我目前正在获得大约5 FPS的帧速率,我希望更快。
我知道pytorch的速度会比tensorRT慢,但我仍然希望速度更快。似乎可能只使用CPU,而不是GPU。
有人可以帮我这个忙吗,或者给我一些见识?
谢谢!