Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/bearpaw/pytorch-classification
/ functions
Functions
98 in github.com/bearpaw/pytorch-classification
⨍
Functions
98
◇
Types & classes
22
↓ 20 callers
Method
update
(self, val, n=1)
utils/misc.py:72
↓ 14 callers
Method
append
(self, numbers)
utils/logger.py:61
↓ 8 callers
Function
gauss
(x,a,b,c)
utils/visualize.py:18
↓ 8 callers
Function
make_layers
(cfg, batch_norm=False)
models/cifar/vgg.py:53
↓ 7 callers
Method
forward
(self, x)
models/cifar/resnext.py:112
↓ 5 callers
Function
make_image
(img, mean=(0,0,0), std=(1,1,1))
utils/visualize.py:12
↓ 4 callers
Method
_make_layer
Stack n bottleneck modules where n is inferred from the depth of the network. Args: block: block type used to construct ResNext
models/imagenet/resnext.py:111
↓ 4 callers
Function
accuracy
Computes the precision@k for the specified values of k
utils/eval.py:5
↓ 4 callers
Method
plot
(self, names=None)
utils/logger.py:70
↓ 3 callers
Method
__init__
(self, depth=22, block=Bottleneck, dropRate=0, num_classes=10, growthRate=12, compressionRate=2)
models/cifar/densenet.py:79
↓ 3 callers
Method
_make_denseblock
(self, block, blocks)
models/cifar/densenet.py:113
↓ 3 callers
Method
_make_layer
(self, block, planes, blocks, stride=1)
models/cifar/preresnet.py:128
↓ 3 callers
Method
_make_layer
(self, block, planes, blocks, stride=1)
models/cifar/resnet.py:129
↓ 3 callers
Method
block
Stack n bottleneck modules where n is inferred from the depth of the network. Args: name: string name of the current block.
models/cifar/resnext.py:92
↓ 3 callers
Method
close
(self)
utils/logger.py:79
↓ 3 callers
Function
savefig
(fname, dpi=None)
utils/logger.py:11
↓ 2 callers
Method
__init__
(self, in_planes, out_planes, stride, dropRate=0.0)
models/cifar/wrn.py:9
↓ 2 callers
Method
__init__
(self, depth, num_classes=1000, block_name='BasicBlock')
models/cifar/preresnet.py:95
↓ 2 callers
Method
__init__
(self, depth, num_classes=1000, block_name='BasicBlock')
models/cifar/resnet.py:95
↓ 2 callers
Method
_make_transition
(self, compressionRate)
models/cifar/densenet.py:122
↓ 2 callers
Function
conv3x3
3x3 convolution with padding
models/cifar/preresnet.py:16
↓ 2 callers
Function
conv3x3
3x3 convolution with padding
models/cifar/resnet.py:16
↓ 2 callers
Function
mkdir_p
make dir if not exist
utils/misc.py:49
↓ 2 callers
Method
set_names
(self, names)
utils/logger.py:47
↓ 2 callers
Function
test
(val_loader, model, criterion, epoch, use_cuda)
imagenet.py:278
↓ 2 callers
Function
test
(testloader, model, criterion, epoch, use_cuda)
cifar.py:286
↓ 1 callers
Method
__init__
Constructor Args: baseWidth: baseWidth for ResNeXt. cardinality: number of convolution groups. layers: co
models/imagenet/resnext.py:75
↓ 1 callers
Method
__init__
Constructor Args: cardinality: number of convolution groups. depth: number of layers. num_classes: number
models/cifar/resnext.py:58
↓ 1 callers
Method
_initialize_weights
(self)
models/cifar/vgg.py:37
↓ 1 callers
Method
_make_layer
(self, block, in_planes, out_planes, nb_layers, stride, dropRate)
models/cifar/wrn.py:38
↓ 1 callers
Function
adjust_learning_rate
(optimizer, epoch)
imagenet.py:336
↓ 1 callers
Function
adjust_learning_rate
(optimizer, epoch)
cifar.py:344
↓ 1 callers
Function
main
()
imagenet.py:110
↓ 1 callers
Function
main
()
cifar.py:101
↓ 1 callers
Function
plot_overlap
(logger, names=None)
utils/logger.py:15
↓ 1 callers
Method
reset
(self)
utils/misc.py:66
↓ 1 callers
Function
save_checkpoint
(state, is_best, checkpoint='checkpoint', filename='checkpoint.pth.tar')
imagenet.py:330
↓ 1 callers
Function
save_checkpoint
(state, is_best, checkpoint='checkpoint', filename='checkpoint.pth.tar')
cifar.py:338
↓ 1 callers
Function
train
(train_loader, model, criterion, optimizer, epoch, use_cuda)
imagenet.py:223
↓ 1 callers
Function
train
(trainloader, model, criterion, optimizer, epoch, use_cuda)
cifar.py:231
Method
__init__
(self)
utils/misc.py:63
Method
__init__
(self, fpath, title=None, resume=False)
utils/logger.py:25
Method
__init__
paths is a distionary with {name:filepath} pair
utils/logger.py:85
Method
__init__
Constructor Args: inplanes: input channel dimensionality planes: output channel dimensionality baseWidth:
models/imagenet/resnext.py:23
Method
__init__
Constructor Args: in_channels: input channel dimensionality out_channels: output channel dimensionality s
models/cifar/resnext.py:19
Method
__init__
(self, inplanes, expansion=4, growthRate=12, dropRate=0)
models/cifar/densenet.py:13
Method
__init__
(self, inplanes, expansion=1, growthRate=12, dropRate=0)
models/cifar/densenet.py:40
Method
__init__
(self, inplanes, outplanes)
models/cifar/densenet.py:62
Method
__init__
(self, nb_layers, in_planes, out_planes, block, stride, dropRate=0.0)
models/cifar/wrn.py:35
Method
__init__
(self, depth, num_classes, widen_factor=1, dropRate=0.0)
models/cifar/wrn.py:47
Method
__init__
(self, num_classes=10)
models/cifar/alexnet.py:13
Method
__init__
(self, features, num_classes=1000)
models/cifar/vgg.py:25
Method
__init__
(self, inplanes, planes, stride=1, downsample=None)
models/cifar/preresnet.py:25
Method
__init__
(self, inplanes, planes, stride=1, downsample=None)
models/cifar/preresnet.py:57
Method
__init__
(self, inplanes, planes, stride=1, downsample=None)
models/cifar/resnet.py:25
Method
__init__
(self, inplanes, planes, stride=1, downsample=None)
models/cifar/resnet.py:57
Function
alexnet
r"""AlexNet model architecture from the `"One weird trick..." <https://arxiv.org/abs/1404.5997>`_ paper.
models/cifar/alexnet.py:39
Function
colorize
Converts a one-channel grayscale image to a color heatmap image
utils/visualize.py:21
Function
densenet
Constructs a ResNet model.
models/cifar/densenet.py:145
Method
forward
(self, x)
models/imagenet/resnext.py:47
Method
forward
(self, x)
models/imagenet/resnext.py:136
Method
forward
(self, x)
models/cifar/resnext.py:42
Method
forward
(self, x)
models/cifar/densenet.py:24
Method
forward
(self, x)
models/cifar/densenet.py:49
Method
forward
(self, x)
models/cifar/densenet.py:69
Method
forward
(self, x)
models/cifar/densenet.py:129
Method
forward
(self, x)
models/cifar/wrn.py:23
Method
forward
(self, x)
models/cifar/wrn.py:43
Method
forward
(self, x)
models/cifar/wrn.py:78
Method
forward
(self, x)
models/cifar/alexnet.py:32
Method
forward
(self, x)
models/cifar/vgg.py:31
Method
forward
(self, x)
models/cifar/preresnet.py:35
Method
forward
(self, x)
models/cifar/preresnet.py:70
Method
forward
(self, x)
models/cifar/preresnet.py:144
Method
forward
(self, x)
models/cifar/resnet.py:35
Method
forward
(self, x)
models/cifar/resnet.py:70
Method
forward
(self, x)
models/cifar/resnet.py:146
Function
get_mean_and_std
Compute the mean and std value of dataset.
utils/misc.py:19
Function
init_params
Init layer parameters.
utils/misc.py:34
Method
plot
(self, names=None)
utils/logger.py:92
Function
preresnet
Constructs a ResNet model.
models/cifar/preresnet.py:160
Function
resnet
Constructs a ResNet model.
models/cifar/resnet.py:162
Function
resnext
Constructs a ResNeXt.
models/cifar/resnext.py:122
Function
resnext101
Construct ResNeXt-101.
models/imagenet/resnext.py:160
Function
resnext152
Construct ResNeXt-152.
models/imagenet/resnext.py:168
Function
resnext50
Construct ResNeXt-50.
models/imagenet/resnext.py:152
Function
show_batch
(images, Mean=(2, 2, 2), Std=(0.5,0.5,0.5))
utils/visualize.py:38
Function
show_mask
(images, masklist, Mean=(2, 2, 2), Std=(0.5,0.5,0.5))
utils/visualize.py:73
Function
show_mask_single
(images, mask, Mean=(2, 2, 2), Std=(0.5,0.5,0.5))
utils/visualize.py:44
Function
vgg11
VGG 11-layer model (configuration "A") Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/cifar/vgg.py:77
Function
vgg11_bn
VGG 11-layer model (configuration "A") with batch normalization
models/cifar/vgg.py:87
Function
vgg13
VGG 13-layer model (configuration "B") Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/cifar/vgg.py:93
Function
vgg13_bn
VGG 13-layer model (configuration "B") with batch normalization
models/cifar/vgg.py:103
Function
vgg16
VGG 16-layer model (configuration "D") Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/cifar/vgg.py:109
Function
vgg16_bn
VGG 16-layer model (configuration "D") with batch normalization
models/cifar/vgg.py:119
Function
vgg19
VGG 19-layer model (configuration "E") Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/cifar/vgg.py:125
Function
vgg19_bn
VGG 19-layer model (configuration 'E') with batch normalization
models/cifar/vgg.py:135
Function
wrn
Constructs a Wide Residual Networks.
models/cifar/wrn.py:88