openvino自己訓練的模型效果不佳? - Cupoy

openvino自己訓練的模型效果不佳?

最近我在嘗試使用openvino進行person reid 在使用 openvino 提供的 model 時效果很好。 然而，當我將自己訓練的model轉為openvino的格式進行測試時效果十分差勁。 想請問我是否少做了甚麼? model : OSNets pretrain model 轉換方式 * 將 pth 轉為 onnx ``` import torch import torchreid from torch.autograd import Variable import torch.onnx datamanager = torchreid.data.ImageDataManager( root='reid-data', sources='market1501', targets='market1501', height=256, width=128, batch_size_train=32, batch_size_test=100, transforms=['random_flip', 'random_crop'] ) model = torchreid.models.build_model( name='osnet_x0_5', num_classes=datamanager.num_train_pids, loss='softmax', pretrained=True ) dummay_input = Variable(torch.randn(1,3,256,128)) checkpoint = torch.load("log/osnet_x0_5/model/model.pth.tar-60") model.load_state_dict(checkpoint['state_dict']) model.eval() torch.onnx.export(model, dummay_input, "osnet_x0_5.onnx") ``` * 之後再進行openvino內建轉換