Как скачать и использовать обученную модель torchvision
Тренировки — это проблема
Простая в использовании модель обученного изображения
Можно использовать различные типы моделей. Есть и относительно новая модель.
Метод
Загрузите и создайте экземпляр предварительно обученной модели.
Вы можете сделать модель только с этим.
Предварительно обработайте входное изображение, чтобы оно соответствовало входу модели.
Измените размер до 224 * 224 и нормализуйте, чтобы он соответствовал набору данных ImageNet.
Пропустите образ через прямой процесс модели.
Результатом является число (1,1000). Надежность соответствует классу ImageNet1000.
Суммируйте значение в % с помощью Softmax.
Загрузите метку класса и отобразите верхнюю часть.
'Египетская кошка' 66.66095733642578
Вы можете проверить модели, которые можно использовать ниже.
https://pytorch.org/vision/master/models.html
🐣
Я инженер-фрилансер.
Консультация по работе
Пожалуйста, не стесняйтесь обращаться к нам с кратким описанием разработки.
[email protected]
Я делаю приложение, использующее Core ML и ARKit.
Мы отправляем информацию, связанную с машинным обучением и дополненной реальностью.