Учебники

PyTorch – функция извлечения в конвентах

Сверточные нейронные сети включают в себя основной признак, извлечение . Следующие шаги используются для реализации функции извлечения сверточной нейронной сети.

Шаг 1

Импортируйте соответствующие модели, чтобы создать модель извлечения объектов с помощью «PyTorch».

import torch
import torch.nn as nn
from torchvision import models

Шаг 2

Создайте класс экстрактора объектов, который можно вызывать по мере необходимости.