cornernet
wnloverforever
这个作者很懒,什么都没留下…
展开
-
Cornernet---decode
我们训练好模型后,如何从网络的输出,decode出所需要的信息呢。也就是模型的训练和测试中的测试部分。首先我们先把网络的输出摆着heatmap_tl : (batch,128,128,80)heatmap_br:(batch,128,128,80)tag_tl : (batch,128,128)tag_br:(batch,128,128)offset_tl:(batch,128,2)...原创 2019-11-22 17:38:54 · 197 阅读 · 1 评论 -
cornernet------loss
首先我们使用网络训练出左上角的heatmap_tl以及右下角的hetmap_br,这个heatmap维度为(batch,128,128,80)其中80是类别数。在制作数据集的时候,将数据也同样制作成了(batch,128,128,80)。以左上角为例,我们每一类都有一个(128,128)的矩阵与之对应。输入网络一张图片,我们会得到左上角以及右下角的heatmap,这个heatmap里面又包括了...原创 2019-11-21 17:05:07 · 497 阅读 · 0 评论 -
Tensorflow----训练
为了充分使用多GPU加速训练,我们使用并联式训练方法(这个名称忘了),因此需要把每一次训练的梯度更新给取出来(这个与训练方法有关)获取所有变量train_variables = tf.trainable_variables()计算梯度tower_grads = []grads = optim.compute_gradients(loss,var_list=train_varia...原创 2019-11-21 10:15:36 · 99 阅读 · 0 评论 -
cornernet---net网络结构(hourglass)
原创 2019-11-20 19:45:57 · 459 阅读 · 0 评论 -
Cornernet-----数据集制作1
我们使用coco数据集来训练我们的cornernet,首先下载数据集,存放路径为D:/dataset\coco\annotations\instances_train2017.json我们主要是使用的.json文件。这里可以参考-----来了解.json文件的存储形式。我们使用from pycocotools.coco import COCO这个工具来读取.json文件1.首先我们要拿到所有...原创 2019-11-19 16:54:10 · 465 阅读 · 0 评论