mask-rcnn-tf2
mask-rcnn-tf2 copied to clipboard
请问老师, 如果我需要将输入shape设置为固定的, 下面几个语句该怎么修改呢?
# input_image = Input(shape=[None, None, config.IMAGE_SHAPE[2]], name="input_image")
input_image = Input(shape=[1024, 1024, config.IMAGE_SHAPE[2]], name="input_image")
#----------------------------------------------#
# meta包含了一些必要信息
#----------------------------------------------#
input_image_meta = Input(shape=[config.IMAGE_META_SIZE],name="input_image_meta")
#----------------------------------------------#
# 输入进来的先验框
#----------------------------------------------#
input_anchors = Input(shape=[None, 4], name="input_anchors")
就是现在我的需求就是想将输入shape设置为1024×1024的, 这个input_anchors = Input(shape=[None, 4], name="input_anchors")中的shape该如何调整呢?
还有如果我如上修改, ,
#----------------------------------------------------------------------#
# 输入的shape大小
# 算法会填充输入图片到[IMAGE_MAX_DIM, IMAGE_MAX_DIM]的大小
#----------------------------------------------------------------------#
"IMAGE_MAX_DIM" : 512,
这里是不是也需要修改成1024呢?
你看注释的话不是只改下面那个就够了吗