DL
mxnet-DL
ht_0628
这个作者很懒,什么都没留下…
展开
-
GluonCV ------ gluoncv/loss.py ------ 常用损失函数
1.SSD损失函数class SSDMultiBoxLoss(gluon.Block): r"""Single-Shot Multibox Object Detection Loss. .. note:: Since cross device synchronization is required to compute batch-wise statistics, it is slightly sub-optimal compared with non原创 2021-09-12 15:01:41 · 78 阅读 · 0 评论 -
GluonCV ------ gluoncv/utils/metrics/voc_detection.py ------ VOC数据集的评估
1.class VOCMApMetric(EvalMetric):1.1 初始化函数 Parameters: --------- iou_thresh : float IOU overlap threshold for TP class_names : list of str optional, if provided, will print out AP for each class """ def __init__(sel原创 2021-09-09 15:13:51 · 131 阅读 · 0 评论 -
Gluoncv -- 训练流程分析和数据结构变化
1. 读取原始数据(以voc为例子)代码分布:gluoncv/data/pascal_voc/detection.py重点类:VOCDetection - 读取标准的VOC数据集;CustomVOCDetection - 读取自定义VOC数据集,可以从xml文件中获取类名;输入数据:VOC类型数据文件的存放路径和部分名称参数。输出数据:采用__item__魔法函数输出 - (图像数据,标签数据) - (H×W×C, N×6) - 其中N为一个图像中标签的个数。2. 加载数据代码分布:mx原创 2021-09-09 15:13:30 · 177 阅读 · 0 评论 -
GluonCV ----- train_ssd.py
1. 解析命令行def parse_args(): parser = argparse.ArgumentParser(description='Train SSD networks.') ##--SSD的主干网络 parser.add_argument('--network', type=str, default='vgg16_atrous', help="Base network name which serves as featu原创 2021-09-07 02:23:23 · 198 阅读 · 0 评论 -
GluonCV ------ anchor.py
@des: 锚框生成器@p1: Index of this generator in SSD models, this is required for naming.@p2: 原图尺寸@p3: 锚框的size@p4: 锚框的ratios@p5: 对应特征图的一个像素相当于原图的step个像素@p6: 通常我们会为大的特征图生成足够的锚,例如128x128。在随后的推断中,我们可以使用可变的输入大小,此时我们可以从这个大型锚图中裁剪相应的锚,这样我们就不用为每个输入重新生成锚。@p7: Cen原创 2021-09-02 23:44:53 · 66 阅读 · 0 评论 -
GluonCV ------ feature.py
1.解析主干网络@des: @p1: 网络类型(str或者自定义的HybridBlock)@p2: 要作为特征提取的层的名称。@p3: 输入的名称。@p4: 预训练 @p5: cpu or gpudef _parse_network(network, outputs, inputs, pretrained, ctx, **kwargs): inputs = list(inputs) if isinstance(inputs, tuple) else inputs for i,原创 2021-09-01 19:53:18 · 104 阅读 · 0 评论 -
GluonCV ------ ssd.py
1.import * 可导入类设置__all__ = ['SSD', 'get_ssd', 'custom_ssd']2.SSD网络类初始化函数@des: 构建一个SSD网络@p1: 主干网络名称(resnet_xx, vgg_xx等等)@p2: 输入图片的大小@p3: 特征提取层的用来作为feature map的stage@p4: 额外卷积层的输出通道数(列表) @p5: 锚框的sizes(列表)@p6: 锚框的ratios(列表)@p7: Step size of anch原创 2021-09-01 10:56:15 · 132 阅读 · 0 评论 -
GluonCV ------ rsenet.py
1. 辅助函数(helpers)def _conv3x3(channels, stride, in_channels): return nn.Conv2D(channels, kernel_size=3, strides=stride, padding=1, use_bias=False, in_channels=in_channels)@des: 封装nn.Conv2D,便捷地生成步幅可控的3×3卷积块。@p1: 输出通道数@p2: 步幅@p3原创 2021-08-31 23:29:26 · 176 阅读 · 0 评论 -
Gluoncv
1.Predict with pre-trained SSD modelsLoad a pretrained modelfrom gluoncv import model_zoo, data, utilsnet = model_zoo.get_model('ssd_512_resnet50_v1_voc', pretrained=True)gluoncv / model_zoo / model_zoo.py...from .ssd import *...def get_model(na原创 2021-08-29 23:11:24 · 183 阅读 · 0 评论 -
Convolutional Neural Network
1. LeNet2. AlexNet3. VGG原创 2021-08-24 12:34:42 · 105 阅读 · 0 评论 -
MXNET_DL_API
9.4 锚框# 生成锚框# data : 输入图像样本(批量大小,通道数,高,宽)# sizes : 锚框大小列表# ratios: 宽高比列表mxnet.contrib.ndarray.MultiBoxPrior(data, sizes, ratios)原创 2021-08-22 10:02:20 · 58 阅读 · 0 评论