Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/datawhalechina/dive-into-cv-pytorch
/ types & classes
Types & classes
77 in github.com/datawhalechina/dive-into-cv-pytorch
⨍
Functions
274
◇
Types & classes
77
↓ 9 callers
Class
Inception
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 callers
Class
VGG
VGG model
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/pytorch-vgg-cifar10/vgg.py:15
↓ 8 callers
Class
VGG
VGG model
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/VGG/vgg_models.py:17
↓ 7 callers
Class
AverageMeter
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 callers
Class
down
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_parts.py:36
↓ 4 callers
Class
up
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_parts.py:49
↓ 3 callers
Class
AverageMeter
Keeps track of most recent, average, sum, and count of a metric.
code/chapter03_object_detection_introduction/tiny_detector_demo/utils.py:669
↓ 3 callers
Class
Embeddings
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:230
↓ 3 callers
Class
Embeddings
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:247
↓ 3 callers
Class
LabelSmoothing
Implement label smoothing.
code/chapter06_transformer/6.2_recognition_by_transformer/train_utils.py:6
↓ 3 callers
Class
LabelSmoothing
Implement label smoothing.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/train_utils.py:39
↓ 3 callers
Class
LayerNorm
Construct a layernorm module (See citation for details).
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:81
↓ 3 callers
Class
LayerNorm
Construct a layernorm module (See citation for details).
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:81
↓ 3 callers
Class
LayerNorm
Construct a layernorm module (See citation for details).
code/chapter06_transformer/6.1_hello_transformer/transformer.py:81
↓ 3 callers
Class
Net
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/NiN/NiN.py:46
↓ 3 callers
Class
SVHNDataset
code/chapter02_image_classification_introduction/2.4_classification_action_SVHN/baseline.py:30
↓ 3 callers
Class
double_conv
(conv => BN => ReLU) * 2
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_parts.py:8
↓ 2 callers
Class
BatchNorm
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/BN/BatchNormalization.py:37
↓ 2 callers
Class
Building_Dataset
code/chapter04_segmentation_introduction/building_identification_baseline/dataset.py:40
↓ 2 callers
Class
Decoder
Generic N layer decoder with masking.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:134
↓ 2 callers
Class
Decoder
Generic N layer decoder with masking.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:151
↓ 2 callers
Class
DecoderLayer
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 callers
Class
DecoderLayer
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 callers
Class
Embeddings
code/chapter06_transformer/6.1_hello_transformer/transformer.py:236
↓ 2 callers
Class
Encoder
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 callers
Class
Encoder
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 callers
Class
EncoderLayer
Encoder is made up of self-attn and feed forward (defined below)
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:113
↓ 2 callers
Class
EncoderLayer
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 callers
Class
Generator
Define standard linear + softmax generation step.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:49
↓ 2 callers
Class
Generator
Define standard linear + softmax generation step.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:49
↓ 2 callers
Class
MultiHeadedAttention
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:184
↓ 2 callers
Class
MultiHeadedAttention
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:201
↓ 2 callers
Class
NoamOpt
Optim wrapper that implements rate.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/train_utils.py:6
↓ 2 callers
Class
PascalVOCDataset
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 callers
Class
PositionalEncoding
Implement the PE function.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:241
↓ 2 callers
Class
PositionalEncoding
Implement the PE function.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:258
↓ 2 callers
Class
PositionwiseFeedForward
Implements FFN equation.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:217
↓ 2 callers
Class
PositionwiseFeedForward
Implements FFN equation.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/transformer.py:234
↓ 2 callers
Class
Recognition_Dataset
code/chapter06_transformer/6.2_recognition_by_transformer/ocr_by_transformer.py:27
↓ 2 callers
Class
Recognition_Dataset
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/ocr_by_transformer.py:29
↓ 2 callers
Class
SimpleLossCompute
A simple loss compute and train function.
code/chapter06_transformer/6.2_recognition_by_transformer/train_utils.py:49
↓ 2 callers
Class
SimpleLossCompute
A simple loss compute and train function.
code/chapter06_transformer/6.2_recognition_by_transformer(online_dataset)/train_utils.py:82
↓ 2 callers
Class
SimpleLossCompute
A simple loss compute and train function.
code/chapter06_transformer/6.1_hello_transformer/first_train_demo.py:147
↓ 2 callers
Class
SublayerConnection
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 callers
Class
SublayerConnection
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 callers
Class
SublayerConnection
实现子层连接结构的类
code/chapter06_transformer/6.1_hello_transformer/transformer.py:95
↓ 1 callers
Class
AlexNet
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/AlexNet/AlexNet.py:48
↓ 1 callers
Class
Batch
Object for holding a batch of data with mask during training.
code/chapter06_transformer/6.1_hello_transformer/first_train_demo.py:26
↓ 1 callers
Class
Decoder
Generic N layer decoder with masking.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:139
↓ 1 callers
Class
DecoderLayer
Decoder is made of self-attn, src-attn, and feed forward (defined below)
code/chapter06_transformer/6.1_hello_transformer/transformer.py:151
↓ 1 callers
Class
Encoder
Encoder The encoder is composed of a stack of N=6 identical layers.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:64
↓ 1 callers
Class
EncoderDecoder
A standard Encoder-Decoder architecture. Base for this and many other models.
code/chapter06_transformer/6.2_recognition_by_transformer/transformer.py:21
↓ 1 callers
Class
EncoderDecoder
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 callers
Class
EncoderDecoder
A standard Encoder-Decoder architecture. Base for this and many other models.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:21
↓ 1 callers
Class
EncoderLayer
EncoderLayer is made up of two sublayer: self-attn and feed forward
code/chapter06_transformer/6.1_hello_transformer/transformer.py:119
↓ 1 callers
Class
Generator
Define standard linear + softmax generation step.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:49
↓ 1 callers
Class
GoogLeNet
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/GoogLeNet/GoogLeNet.py:66
↓ 1 callers
Class
LabelSmoothing
Implement label smoothing.
code/chapter06_transformer/6.1_hello_transformer/first_train_demo.py:123
↓ 1 callers
Class
ModelSaver
code/chapter04_segmentation_introduction/building_identification_baseline/utils/model_saver.py:17
↓ 1 callers
Class
MultiBoxLoss
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 callers
Class
MultiHeadedAttention
code/chapter06_transformer/6.1_hello_transformer/transformer.py:189
↓ 1 callers
Class
Net
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/ResNet/ResNet.py:75
↓ 1 callers
Class
Net
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/lenet/LeNet.py:46
↓ 1 callers
Class
Net
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/BN/BatchNormalization.py:99
↓ 1 callers
Class
Net
code/chapter01_preliminary_knowledge/1.5_FC_MNIST_Classification/FC_MNIST_Classification.py:13
↓ 1 callers
Class
OCR_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 callers
Class
OCR_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 callers
Class
PositionalEncoding
Implement the PE function.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:248
↓ 1 callers
Class
PositionwiseFeedForward
Implements FFN equation.
code/chapter06_transformer/6.1_hello_transformer/transformer.py:223
↓ 1 callers
Class
PredictionConvolutions
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 callers
Class
SVHN_Model1
code/chapter02_image_classification_introduction/2.4_classification_action_SVHN/baseline.py:100
↓ 1 callers
Class
UNet
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_model.py:7
↓ 1 callers
Class
VGGBase
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 callers
Class
inconv
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_parts.py:26
↓ 1 callers
Class
outconv
code/chapter04_segmentation_introduction/building_identification_baseline/unet/unet_parts.py:81
↓ 1 callers
Class
tiny_detector
The tiny_detector network 包含一个VGG作为特征提取模块,并在最后一个特征图上添加一个输出头来预测目标框信息
code/chapter03_object_detection_introduction/tiny_detector_demo/model.py:168
Class
ResidualBlock
code/chapter02_image_classification_introduction/2.2_introduction_of_image_classification/classical_cnn_models/ResNet/ResNet.py:49