LiYangDoYourself

固话成pb模型时出现的问题

8

你好，我训练完模型之后，将元文件，固话成PB文件时，做C ++调用，在pnet模型由于固定了输入图片的BATCHSIZE和大小，导致不能输入任意其他大小的图片，我尝试将输入改为（None，None，None，3），但是出现计算分类损失时需要这个BATCHSIZE，所以我在定义的时候这个BATCHSIZE就不能为空，但我必须这样定义（None，None，None，3），我的后续pb模型才能输入一张任意大小的图片。