bubbliiiing / mask-rcnn-tf2

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

请问老师, 如果我需要将输入shape设置为固定的, 下面几个语句该怎么修改呢? #23

Open stu-zhangxl opened 1 year ago

stu-zhangxl commented 1 year ago
# 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该如何调整呢?

stu-zhangxl commented 1 year ago

还有如果我如上修改, ,

#----------------------------------------------------------------------#
#   输入的shape大小
#   算法会填充输入图片到[IMAGE_MAX_DIM, IMAGE_MAX_DIM]的大小
#----------------------------------------------------------------------#
"IMAGE_MAX_DIM"     : 512,

这里是不是也需要修改成1024呢?

bubbliiiing commented 1 year ago

你看注释的话不是只改下面那个就够了吗