Я работаю с моделью глубокого обучения, которая имеет Re sNet -50 в качестве магистрали, предварительно обученной на Imag eNet. Я использую набор данных CUB-200, который представляет собой набор из 200 видов птиц. По этой причине я думаю, что было бы неплохо иметь предварительно обученную модель для набора данных, имеющего аналогичную область, и я обнаружил, что iNaturalist может быть той, которую я ищу.
Проблема в том, что я не нашел ни одной предварительно обученной модели для Pytorch, а только Tensorflow здесь .
Я попытался преобразовать его с помощью библиотеки MDNN, но для этого требуется также расширение файла .ckpt.meta, и у меня есть только «.ckpt».
Это пример использования библиотеки MDNN для преобразования модели tf в факел:
mmconvert -sf tensorflow -in imagenet_resnet_v2_152.ckpt.meta -iw imagenet_resnet_v2_152.ckpt --dstNode MMdnn_Output -df pytorch -om tf_to_pytorch_resnet_152.pth
Может ли кто-нибудь помочь мне с этим?