我已经得到了一个pytorch模型文件，和一些对象检测结果。对象检测结果给出了数字，以确定它检测到什么样的对象，但我想从模型文件的名称。
我找到的一些python代码如下所示
model = DetectMultiBackend(weights, device=device, dnn=dnn, data=data, fp16=half)
stride, names, pt = model.stride, model.names, model.pt我很确定我需要得到names数组，但是我用的是java，不是python。我查看了ai.djl.pytorch.engine.PtModel，但是没有发现任何看起来像是从数字到名称的Map。
看起来DeepJavaLibrary甚至不支持加载一个普通的.pt文件：
第一个
使用Java和PyTorch模型文件将对象/类编号Map到名称的正确方法是什么？
1条答案
按热度按时间jhiyze9q1#
问：从pyTorch模型文件Map到Java对象的正确方法是什么？
答：我认为pyTorch模型文件只是Python对象的pickle序列化，一个替代方案可能是PythonPickle。
您的挑战是读取Java中的PyTorch（.pt？）模型文件。
听起来你已经开始走这条路了，并且取得了一些成功。如果这对你有用的话-太好了！
无论如何-请让我们知道你的决定。