Obtain features from ArcFace - vision - PyTorch Forums
discuss.pytorch.org › t › obtain-features-fromMay 01, 2020 · I’m trying to extract features using pytorch implementation of ArcFace Here is my code: import numpy as np from tqdm import tqdm import torch import torch.nn as nn import torchvision.transforms as transforms from torchvision import datasets from model import Backbone transform = transforms.Compose([ transforms.RandomHorizontalFlip(), transforms.ToTensor(), transforms.Normalize(mean=[0.5, 0.5 ...