lscelory的博客

这里写自定义目录标题DeeplabV3 ResNet101调用torchvision.models.segmentation源码接口的定义函数deeplabv3_resnet101加载模型的函数_load_model创建用于分割的resnet函数_segm_resnettorchvision.models源码torchvision.models.resnet.pytorchvision.model...

二、Mask R-CNN代码解读FAIR在发布detectron的同时，也发布了一系列的tutorial文件，接下来将根据Detectron/GETTING_STARTED.md文件来解读代码。先来看一下detectron的文件结构。config中是训练和测试的配置文件，官方的baseline的参数配置都以.yaml文件的形式存放在其中。 demo一些图像示例还有分割好的结果。...

Mask R-CNN的源码是随着FAIR的物体检测平台detectron公布的，detectron囊括了从Fast R-CNN到Mask R-CNN所有重要的物体检测算法，整个平台基于caffe2实现。本文主要从代码层面介绍Mask R-CNN具体的实现过程。一、caffe2介绍caffe2是在caffe的基础上的进化版，最大的特色就是引入了operator的概念，取代了caffe中lay...

DeepLabv3+代码阅读之train.py一、main()def main(unused_argv):# main必须带参数，否则报错：'TypeError: main() takes no arguments (1 given)'; # main的参数名随意定义，无要求。 tf.logging.set_verbosity(tf.loggi...

DeepLabv3+代码阅读之data_generator.py一、Dataset1. __init__class Dataset(object): """Represents input dataset for deeplab model.""" def __init__(self, dataset_name,# Dataset name ...

DeepLabv3+代码阅读之input_preprocess.py一、preprocess_image_and_label对图片和标签处理返回： original_image： 原始图片 processed_image： 处理后图片 label： 处理后ground truthdef preprocess_image_and_label(image,# 输入图片 ...

DeepLabv3+代码阅读之common.py一、ModelOptionsclass ModelOptions( collections.namedtuple('ModelOptions', [ 'outputs_to_num_classes',# 输出类型对应类别数的字典，会得到如：outputs_to_num_classes['semantic'] = 21 ...

DeepLabv3+代码阅读之model.py一、multi_scale_logits多尺度输入，得到输出概率参数： images: 输出图片，尺寸 [batch, height, width, channels]. model_options: 配置模型的参数选择，一个ModelOptions实例 image_pyramid: 图片金字塔，输入图片的多个尺度，如果没有此项输入，则为[...

DeepLabv3+代码阅读之train_utils.py一、get_model_learning_rate()def get_model_learning_rate(learning_policy,# Learning rate policy for training. base_learning_rate,# The base lea...