实时图像分割
lbai7134
工程师
展开
-
TorchSeg代码学习笔记(二:data和augmentation)
本文以cityScapes和BiSeNet.R18为例BaseDatasetTorchSeg的dataset定义在furnace/datasets/BaseDataset.py中,根据setting来提供基本的图像和label的读取功能。而setting定义在各个网络的config.py中定义。cityscapes数据集cityScapes数据集的定义在furnace/datasets/cityscapes/cityscapes.pyimport numpy as npfrom data原创 2020-11-13 05:49:54 · 452 阅读 · 1 评论 -
TorchSeg代码学习笔记(一:安装)
代码repo:https://github.com/ycszen/TorchSegTorchSeg安装:由于本机上没有安装nvcc,我采用docker来实现下载对应docker:docker pull nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04运行容器:docker run --gpus all -p 12222:22 -it --name=lbaiSeg --mount type=bind,source=/mnt/VOL_2/lbai/,target=/l原创 2020-11-12 11:47:36 · 240 阅读 · 0 评论