Как преобразовать предварительно обученную модель в модель классификации изображений CoreML для использования на iOS
Мы хотим преобразовать модель классификации изображений
Если вы преобразуете модель классификации изображений TorchVision как есть, на выходе будет числовое значение 1000.
Мы хотели бы преобразовать его в процент достоверности, чтобы его можно было предварительно просмотреть как модель классификации изображений.
Добавить имя класса и слой softmax
Добавьте имя класса и слой softmax, чтобы получить процент достоверности для каждого класса в 1000 классов, и вы также можете использовать предварительный просмотр xcode.
Метод
Скачайте модель с Torchvision.
Создайте класс модели-оболочки с добавленным слоем softmax для отображения в процентах.
Загрузите метку класса ImageNet, создайте список и создайте конфигурацию классификатора CoreML из списка.
преобразование.
Теперь вы можете просмотреть его в Xcode.
Как использовать на iOS
королевский пингвин 0,61…
🐣
Я инженер-фрилансер.
Консультация по работе
Пожалуйста, не стесняйтесь обращаться к нам с кратким описанием разработки.
[email protected]
Я делаю приложение, использующее Core ML и ARKit.
Мы отправляем информацию, связанную с машинным обучением и дополненной реальностью.