Как я могу использовать модель формата .mat в PyTorch?

Question

Я загрузил обученную модель NetVLAD с https://www.di.ens.fr/willow/research/netvlad/ . Однако модель была обучена в MatLab и имеет тип .mat . Как я могу загрузить это в PyTorch?

Мне удалось получить доступ к весам, используя следующий код, но я не знаю, как действовать дальше. Есть ли способ загрузить это в PyTorch с помощью load_state_dict или чего-то подобного?

      from scipy.io import loadmat

data = loadmat('vd16_pitts30k_conv5_3_vlad_preL2_intra_white.mat')

array_data = data['net']

Answer 1

Веса не имеют никакой информации о структуре сети, это просто массив чисел. Если у вас точно такая же сеть, как у MATLAB, и вы знаете, куда поместить каждое число, да, вы можете это сделать, но «прямого и простого» способа не существует.