caffe学习
tang-0203
这个作者很懒,什么都没留下…
展开
-
windows下caffe-ssd中resize的坑
问题描述在windows平台上,本地训练SSD_512得到了对应的权值参数文件,加载模型进行前向测试的时候,发现调用caffe.io.Transformer中的resize处理函数速度太慢,打算用opencv的resize做替换,因此更改了输入图片到模型中的预处理过程,使用caffe.io.Transformer进行预处理的过程如下:import numpy as np import sys,o原创 2017-12-28 09:28:16 · 1590 阅读 · 0 评论 -
nvidia显卡查看Compute Capability
编译caffe的时候,Makefile.config文件需要指定gencode arch,这个时候可以去nvidia官网查看自己显卡对应的Compute Capability. 附上链接: https://developer.nvidia.com/cuda-gpus原创 2018-05-01 11:17:25 · 6900 阅读 · 0 评论