![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
图像分割代码
蓝德库洛尔多
这个作者很懒,什么都没留下…
展开
-
pytorch-semseg源码解读test.py
这部分代码很坑,原作者代码里若不更改命令行参数norm,则会进行两次标准化 import os import torch import argparse import numpy as np import scipy.misc as misc from ptsemseg.models import get_model from ptsemseg.loader import get_loader from ptsemseg.utils import convert_state_dict try:原创 2020-06-10 22:12:28 · 883 阅读 · 2 评论 -
pytorch-semseg源码解析cityscapes_loader.py
关于[::-1],[:, ::-1],[:, :, ::-1]的区别: from PIL import Image import imageio import numpy as np img = np.arange(48).reshape(4,4,3) print("no.1",img[::-1]) print("no.2",img[:,::-1]) print("no.3",img[:,:,::-1]) 输出长这样: no.1 [[[36 37 38] [39 40 41] [42原创 2020-06-09 20:01:11 · 846 阅读 · 0 评论 -
pytorch-semseg代码解读分割loss.py
其中关于contiguous()函数介绍在: PyTorch中的contiguous 其中关于transpose()函数在stack overflow上有个问题讲的不错: Tranpose讲解 关于input=input.transpose(1,2).transpose(2,3).contiguous().view(-1,c)的详细解释: 比如假如原来input 是 2*3*10*10,也就是两张图,每张图三个通道,像素大小为10*10=100个,第一次transpose(1,2)之后,...原创 2020-06-08 20:38:06 · 839 阅读 · 0 评论 -
pytoch-semseg代码解析学习率调整scheduler.py
from torch.optim.lr_scheduler import _LRScheduler class ConstantLR(_LRScheduler): def __init__(self, optimizer, last_epoch=-1): super(ConstantLR, self).__init__(optimizer, last_epoch)# 调用_LRScheduler的init初始化 def get_lr(self): re.原创 2020-06-07 21:22:28 · 367 阅读 · 0 评论 -
pytorch-semseg图像分割代码解读utils.py
""" Misc Utility functions """ import os import logging import datetime import numpy as np from collections import OrderedDict def recursive_glob(rootdir=".", suffix=""): """Performs recursive glob with given suffix and rootdir :param rootd.原创 2020-06-07 20:41:30 · 708 阅读 · 0 评论