Functions — github.com/KaihuaTang/ResNet50-Pytorch-Face-Recognition

CodeHub MCPcopy Create free account

hub / github.com/KaihuaTang/ResNet50-Pytorch-Face-Recognition / functions

Functions17 in github.com/KaihuaTang/ResNet50-Pytorch-Face-Recognition

⨍ Functions 17 ◇ Types & classes 5

↓ 4 callersMethodmake_stack

(self, planes, blocks, stride = 1)

↓ 1 callersMethod__init__

(self, layers, num_classes=1000, model_path="model.pkl")

↓ 1 callersMethodinit_param

↓ 1 callersMethodinit_param

↓ 1 callersFunctionresnet50

Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet

↓ 1 callersMethodstart_train

Detail of training

↓ 1 callersFunctionvgg16

(pretrained = False, **kwargs)

Method__getitem__

Get the image item by index

Init Dataset, Model and others

(self, number_classes = 2000, model_path="model.pkl")

Initialize some variables Load labels & names define transform

(self, inplanes, planes, stride=1, downsample=None)

Get the length of the entire dataset

Functionresnet101

Constructs a ResNet-101 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet