MCPcopy Create free account

hub / github.com/datawhalechina/dive-into-cv-pytorch / types & classes

Types & classes77 in github.com/datawhalechina/dive-into-cv-pytorch

↓ 9 callersClassInception
1、输入通过Inception模块的4个分支分别计算,得到的输出宽和高相同(因为使用了padding),而通道不同。 2、将4个分支的通道进行简单的合并,即得到Inception模块的输出。 3、每次卷积之后都使用批正则化`BatchNorm2d`,并使用relu函数进行激
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/GoogLeNet/GoogLeNet.py:10
↓ 8 callersClassVGG
VGG model
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/pytorch-vgg-cifar10/vgg.py:15
↓ 8 callersClassVGG
VGG model
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/VGG/vgg_models.py:17
↓ 7 callersClassAverageMeter
Computes and stores the average and current value
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/pytorch-vgg-cifar10/main.py:264
↓ 4 callersClassdown
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_parts.py:36
↓ 4 callersClassup
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_parts.py:49
↓ 3 callersClassAverageMeter
Keeps track of most recent, average, sum, and count of a metric.
code/chapter03_object_detection_introduction/tiny_detector_demo/utils.py:669
↓ 3 callersClassEmbeddings
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:230
↓ 3 callersClassEmbeddings
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:247
↓ 3 callersClassLabelSmoothing
Implement label smoothing.
code/chapter06_transformer/6.2_recognition_by_transformer/train_utils.py:6
↓ 3 callersClassLabelSmoothing
Implement label smoothing.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/train_utils.py:39
↓ 3 callersClassLayerNorm
Construct a layernorm module (See citation for details).
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:81
↓ 3 callersClassLayerNorm
Construct a layernorm module (See citation for details).
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:81
↓ 3 callersClassLayerNorm
Construct a layernorm module (See citation for details).
code/chapter06_transformer/6.1_hello_transformer/transformer.py:81
↓ 3 callersClassNet
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/NiN/NiN.py:46
↓ 3 callersClassSVHNDataset
code/chapter02_image_classification_introduction/2.4_classification_action_SVHN/baseline.py:30
↓ 3 callersClassdouble_conv
(conv => BN => ReLU) * 2
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_parts.py:8
↓ 2 callersClassBatchNorm
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/BN/BatchNormalization.py:37
↓ 2 callersClassBuilding_Dataset
code/chapter04_segmentation_introduction/building_identification_baseline/dataset.py:40
↓ 2 callersClassDecoder
Generic N layer decoder with masking.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:134
↓ 2 callersClassDecoder
Generic N layer decoder with masking.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:151
↓ 2 callersClassDecoderLayer
Decoder is made of self-attn, src-attn, and feed forward (defined below)
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:146
↓ 2 callersClassDecoderLayer
Decoder is made of self-attn, src-attn, and feed forward (defined below)
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:163
↓ 2 callersClassEmbeddings
code/chapter06_transformer/6.1_hello_transformer/transformer.py:236
↓ 2 callersClassEncoder
Encoder The encoder is composed of a stack of N=6 identical layers.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:64
↓ 2 callersClassEncoder
Encoder The encoder is composed of a stack of N=6 identical layers.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:64
↓ 2 callersClassEncoderLayer
Encoder is made up of self-attn and feed forward (defined below)
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:113
↓ 2 callersClassEncoderLayer
Encoder is made up of self-attn and feed forward (defined below)
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:130
↓ 2 callersClassGenerator
Define standard linear + softmax generation step.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:49
↓ 2 callersClassGenerator
Define standard linear + softmax generation step.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:49
↓ 2 callersClassMultiHeadedAttention
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:184
↓ 2 callersClassMultiHeadedAttention
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:201
↓ 2 callersClassNoamOpt
Optim wrapper that implements rate.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/train_utils.py:6
↓ 2 callersClassPascalVOCDataset
A PyTorch Dataset class to be used in a PyTorch DataLoader to create batches.
code/chapter03_object_detection_introduction/tiny_detector_demo/datasets.py:9
↓ 2 callersClassPositionalEncoding
Implement the PE function.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:241
↓ 2 callersClassPositionalEncoding
Implement the PE function.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:258
↓ 2 callersClassPositionwiseFeedForward
Implements FFN equation.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:217
↓ 2 callersClassPositionwiseFeedForward
Implements FFN equation.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:234
↓ 2 callersClassRecognition_Dataset
code/chapter06_transformer/6.2_recognition_by_transformer/ocr_by_transformer.py:27
↓ 2 callersClassRecognition_Dataset
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/ocr_by_transformer.py:29
↓ 2 callersClassSimpleLossCompute
A simple loss compute and train function.
code/chapter06_transformer/6.2_recognition_by_transformer/train_utils.py:49
↓ 2 callersClassSimpleLossCompute
A simple loss compute and train function.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/train_utils.py:82
↓ 2 callersClassSimpleLossCompute
A simple loss compute and train function.
code/chapter06_transformer/6.1_hello_transformer/first_train_demo.py:147
↓ 2 callersClassSublayerConnection
A residual connection followed by a layer norm. Note for code simplicity the norm is first as opposed to last.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:95
↓ 2 callersClassSublayerConnection
A residual connection followed by a layer norm. Note for code simplicity the norm is first as opposed to last.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:111
↓ 2 callersClassSublayerConnection
实现子层连接结构的类
code/chapter06_transformer/6.1_hello_transformer/transformer.py:95
↓ 1 callersClassAlexNet
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/AlexNet/AlexNet.py:48
↓ 1 callersClassBatch
Object for holding a batch of data with mask during training.
code/chapter06_transformer/6.1_hello_transformer/first_train_demo.py:26
↓ 1 callersClassDecoder
Generic N layer decoder with masking.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:139
↓ 1 callersClassDecoderLayer
Decoder is made of self-attn, src-attn, and feed forward (defined below)
code/chapter06_transformer/6.1_hello_transformer/transformer.py:151
↓ 1 callersClassEncoder
Encoder The encoder is composed of a stack of N=6 identical layers.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:64
↓ 1 callersClassEncoderDecoder
A standard Encoder-Decoder architecture. Base for this and many other models.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:21
↓ 1 callersClassEncoderDecoder
A standard Encoder-Decoder architecture. Base for this and many other models.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:21
↓ 1 callersClassEncoderDecoder
A standard Encoder-Decoder architecture. Base for this and many other models.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:21
↓ 1 callersClassEncoderLayer
EncoderLayer is made up of two sublayer: self-attn and feed forward
code/chapter06_transformer/6.1_hello_transformer/transformer.py:119
↓ 1 callersClassGenerator
Define standard linear + softmax generation step.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:49
↓ 1 callersClassGoogLeNet
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/GoogLeNet/GoogLeNet.py:66
↓ 1 callersClassLabelSmoothing
Implement label smoothing.
code/chapter06_transformer/6.1_hello_transformer/first_train_demo.py:123
↓ 1 callersClassModelSaver
code/chapter04_segmentation_introduction/building_identification_baseline/utils/model_saver.py:17
↓ 1 callersClassMultiBoxLoss
The loss function for object detection. 对于Loss的计算,完全遵循SSD的定义,即 MultiBox Loss This is a combination of: (1) a localization loss for t
code/chapter03_object_detection_introduction/tiny_detector_demo/model.py:335
↓ 1 callersClassMultiHeadedAttention
code/chapter06_transformer/6.1_hello_transformer/transformer.py:189
↓ 1 callersClassNet
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/ResNet/ResNet.py:75
↓ 1 callersClassNet
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/lenet/LeNet.py:46
↓ 1 callersClassNet
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/BN/BatchNormalization.py:99
↓ 1 callersClassNet
code/chapter01_preliminary_knowledge/1.5_FC_MNIST_Classification/FC_MNIST_Classification.py:13
↓ 1 callersClassOCR_EncoderDecoder
A standard Encoder-Decoder architecture. Base for this and many other models.
code/chapter06_transformer/6.2_recognition_by_transformer/ocr_by_transformer.py:145
↓ 1 callersClassOCR_EncoderDecoder
A standard Encoder-Decoder architecture. Base for this and many other models.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/ocr_by_transformer.py:132
↓ 1 callersClassPositionalEncoding
Implement the PE function.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:248
↓ 1 callersClassPositionwiseFeedForward
Implements FFN equation.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:223
↓ 1 callersClassPredictionConvolutions
Convolutions to predict class scores and bounding boxes using feature maps. The bounding boxes (locations) are predicted as encoded offsets
code/chapter03_object_detection_introduction/tiny_detector_demo/model.py:102
↓ 1 callersClassSVHN_Model1
code/chapter02_image_classification_introduction/2.4_classification_action_SVHN/baseline.py:100
↓ 1 callersClassUNet
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_model.py:7
↓ 1 callersClassVGGBase
VGG base convolutions to produce feature maps. 完全采用vgg16的结构作为特征提取模块,丢掉fc6和fc7两个全连接层。 因为vgg16的ImageNet预训练模型是使用224×224尺寸训练的,因此我们的网络输入也固定为22
code/chapter03_object_detection_introduction/tiny_detector_demo/model.py:10
↓ 1 callersClassinconv
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_parts.py:26
↓ 1 callersClassoutconv
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_parts.py:81
↓ 1 callersClasstiny_detector
The tiny_detector network 包含一个VGG作为特征提取模块,并在最后一个特征图上添加一个输出头来预测目标框信息
code/chapter03_object_detection_introduction/tiny_detector_demo/model.py:168
ClassResidualBlock
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/ResNet/ResNet.py:49