![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
SceneClassification
文章平均质量分 53
书旅小组的黄叫兽
这个作者很懒,什么都没留下…
展开
-
深度学习应该选择tensorflow还是pytorch?一个场景分类任务惹出的问题
最近在做一个场景分类的任务,用AlexNet在Places数据集上做场景的分类训练,使用的框架是TF,自己不到2G的显存很容易就崩溃了,一直出现如下错误:ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[12544,4096] [[Nod...原创 2018-05-17 22:59:51 · 2584 阅读 · 0 评论 -
[Learning Deep Features for Scene Recognition using Places Database]场景分类demo的实现
参考:https://github.com/CSAILVision/places365 论文:Learning Deep Features for Scene Recognition using Places Database需要预先安装包:numpypytorchopencv-pythonPillow首先将places365的代码git到本地:git clone ht...原创 2018-05-18 20:54:39 · 1570 阅读 · 1 评论 -
[菜鸟向] Pytorch的模型与变量由CPU模式转GPU模式的操作与debug
1.修改CPU版本的pytorch模型到GPU版本从github上git下来的CNNs程序需要将分类结果在opencv上显示,图片是采用单帧处理,因此每一帧的图片返回类别与置信度的速度需要加快,于是想到该CPU版本的模型到GPU版本。在参考网上的文档和博客后,发现只要在模型和参数后加.cuda()即可将模型由cpu上的运算调到gpu上运算。首先需要确定自己的pytorch版本能否进行g...原创 2018-05-22 19:17:32 · 11752 阅读 · 3 评论