Сверточные нейронные сети включают в себя основной признак, извлечение . Следующие шаги используются для реализации функции извлечения сверточной нейронной сети.
Шаг 1
Импортируйте соответствующие модели, чтобы создать модель извлечения объектов с помощью «PyTorch».
import torch import torch.nn as nn from torchvision import models
Шаг 2
Создайте класс экстрактора объектов, который можно вызывать по мере необходимости.