摘要: 环境依赖: pytorch 0.4以上 tensorboardX: pip install tensorboardX、pip install tensorflow 在项目代码中加入tensorboardX的记录代码,生成文件并返回到浏览器中显示可视化结果。 官方示例: 默认设置是在根目录下生成一个r 阅读全文
posted @ 2019-04-15 19:42 you-wh 阅读(4034) 评论(0) 推荐(0) 编辑
摘要: 前提:ubuntu+tensorflow-gpu+python3.6 各种环境提前配好 1.下载工程源码 网址:https://github.com/tensorflow/models 下载时会遇到速度过慢或中间因为网络错误停止,可以换移动网络或者用迅雷下载。 2.测试环境 先添加slim路径,每次 阅读全文
posted @ 2019-04-15 17:04 you-wh 阅读(2365) 评论(0) 推荐(0) 编辑
摘要: 步骤一: 找到一个CNN分类模型 步骤二: 对于这个模型修改最后的分类层,为21类,去掉最后一个全连接层 步骤三: 根据选择性搜索提取图像的候选区域(框) 对于候选框修改大小以适应CNN输入,然后得出feature map即为提取到的特征 步骤四: 训练SVM分类器,对特定敏感区域进行分类,每个类别 阅读全文
posted @ 2019-04-15 17:02 you-wh 阅读(753) 评论(0) 推荐(0) 编辑
摘要: pytorch快速加载预训练模型参数的方式 针对的预训练模型是通用的模型,也可以是自定义模型,大多是vgg16 , resnet50 , resnet101 , 等,从官网加载太慢 直接修改源码,改为本地地址 1.直接使用默认程序里的下载方式,往往比较慢; 2.通过修改源代码,使得模型加载已经下载好 阅读全文
posted @ 2019-04-15 17:02 you-wh 阅读(5312) 评论(0) 推荐(1) 编辑
摘要: 将训练好的语义分割模型保存下来,重新加载之后 通过这一个操作得到标签; 绘图的主函数在下面: 阅读全文
posted @ 2019-04-15 15:58 you-wh 阅读(2511) 评论(0) 推荐(0) 编辑
Fork me on GitHub