视觉检测
大豆虫
踏上神经之旅
展开
-
K.image_dim_ordering和K.image_data_format
keras2.3.1self.dim_ordering = K.image_data_format()assert self.dim_ordering in {‘channels_first’, ‘channels_last’}调整图像的维度顺序,first通道在前(3,416,416),last通道在后(416,416,3)keras1.xself.dim_ordering = K.image_dim_ordering()assert self.dim_ordering in {‘th’, ‘原创 2020-10-18 19:11:22 · 711 阅读 · 0 评论 -
笔记add_weight() got multiple values for argument ‘name‘
Faster_RCNN解决add_weight() got multiple values for argument ‘name’最近在根据学习视觉检测的时候,在github上下载的代码运行时报错“add_weight() got multiple values for argument ‘name’”,小白的我各种尝试,发现把add_weight()中的shape,改成shape=shape就不报错了,这是啥子机理嘛...原创 2020-10-18 18:53:48 · 2091 阅读 · 0 评论 -
笔记Hint: If you want to see a list of allocated tensors when OOM
Faster-rcnn 爆存问题前言在根据睿智博主大佬的检测算法复现自己的数据集时,发现哎呀,内存不够啦Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.一、解决方法?我发现把self.num_rois改小就可以跑起来啦,不知道这种解决方法有理论依原创 2020-10-18 18:39:40 · 713 阅读 · 0 评论