Skip to content

这份code是不是不支持多卡训练?另外还有一些错误。 #3

@stoneforestwhu

Description

@stoneforestwhu

def build_mrcnn_head()返回值mrcnn_bbox的shape是[?,?,8],但是build_mrcnn_bbox_loss()的输入mrcnn_bbox的shape是[?,?,2,4],shape不匹配。此处并不会报错,但是逻辑错了。
def build_rescore_graph()中计算box_level时采用的坐标是归一化坐标,因此需要乘以1024。但是计算instance_score_0时所有的box得到的score都是一样的,因为reduce_mean()操作之后,张量的shape变为1了。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions