语义分割
文章平均质量分 56
语义分割算法学习与实践
三月微暖寻春笋
选择大于努力,方向决定努力的价值。所有的努力与挣扎都是为了内心的平和。
展开
-
pytorch框架下训练自己的数据集【二】
踩过的坑:RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED实际类别数目与设置的类别数目不一致导致。原创 2021-04-01 12:30:34 · 997 阅读 · 4 评论 -
pytorch训练的.pth模型格式转换
深度网络模型再工程化调用时存在模型不识别的问题,主要时推理的程序有不同的语言编写的,本文介绍使用pytorch训练的模型格式.pth转化为.pt格式的代码。pytorch网络代码地址:https://github.com/yassouali/pytorch_segmentation,安装请参考《pytorch框架下语义分割训练实践(一)》。1 .pth模型转换为.pt模型我是用fcn训练得到一个语言分割模型,checkpoint-epoch100.pth,使用inference.py文件可以正.原创 2021-03-30 12:21:30 · 8102 阅读 · 9 评论 -
语义分割制作自己的voc数据集
1.单文件转换使用conda安装完后,在conda安装目录下有如下几个可执行程序,使用labelme_json_to_dataset.exe可以转换单个json文件,但不能批量转换。输入以下命令进行单个文件转换:labelme_json_to_dataset.exe json文件路径2.批处理程序在conda安装目录中,进入到如下目录:"你的conda安装路径\Lib\site-packages\labelme\cli\",现将原先的json_to_dataset.py文件备份,将原创 2020-09-16 11:28:50 · 2985 阅读 · 1 评论 -
pytorch框架下语义分割训练实践(一)
源码地址:https://github.com/yassouali/pytorch_segmentation环境准备torch==1.1.0 torchvision==0.3.0tqdm==4.32.2tensorboard==1.14.0Pillow==6.2.0opencv-python==4.1.0.25这里面几个只有torch比较大,其他都很小,很快就装完,安装库前务必装下pip,python的话尽量使用3.6及以上版本。比如安装tensorboard...原创 2020-09-15 09:41:01 · 1649 阅读 · 11 评论 -
pytorch和torchvision安装
使用pip安装pytorch和torchvision,如未安装pip工具,则需先安装pip,安装命令:sudo apt install python-pip国内由于网速的限制,使用命令安装时需在后面加上清华的镜像,这样下载速度会大大提升,安装更容易成功。pip install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple/此命令在安装torchvision的同时,会把torch一起安装。安装结束后重新开启一个终端并进入py原创 2020-09-09 13:45:52 · 41023 阅读 · 10 评论