bubbliiiing / mask-rcnn-tf2

这是一个mask-rcnn-tf2的库,可以用于训练自己的模型。
MIT License
64 stars 9 forks source link

这段为啥要重复添加一次? #35

Open Sokkafan opened 1 year ago

Sokkafan commented 1 year ago

------------------------------------------------------------------

获得RPN网络的预测结果,进行格式调整,把五个特征层的结果进行堆叠

------------------------------------------------------------------

rpn_class_logits, rpn_class, rpn_bbox = [],[],[]
for p in rpn_feature_maps:
    logits,classes,bbox = rpn([p])
    rpn_class_logits.append(logits)
    rpn_class.append(classes)
    rpn_bbox.append(bbox)

------------------------------------------------------------------

此时获得的rpn_class_logits、rpn_class、rpn_bbox的维度是

rpn_class_logits : Batch_size, num_anchors, 2

rpn_class : Batch_size, num_anchors, 2

rpn_bbox : Batch_size, num_anchors, 4

------------------------------------------------------------------

for p in rpn_feature_maps: logits,classes,bbox = rpn([p]) rpn_class_logits.append(logits) rpn_class.append(classes) rpn_bbox.append(bbox)

Sokkafan commented 1 year ago

之前出现过 #26 类似的问题,把这段重复的删掉以后就好了。 @bubbliiiing

bubbliiiing commented 1 year ago

好像是错了,我的问题………………我改一下