跑inception V3模型时,出现报错
报错:
terminate called after throwing an instance of 'dmlc::Error'
what(): [10:06:15] src/io/image_aug_default.cc:282: Check failed: static_cast<index_t>(res.rows) >= param_.data_shape[1] && static_cast<index_t>(res.cols) >= param_.data_shape[2] input image size smaller than input shape
原因:
image_aug_default说明是数据增强导致输入图像的size小于模型的input shape,
解决:
注释
#data.set_data_aug_level(parser, 3)
#min_random_scale = 0.533
或
inception v3的输入图像为299*299,因此增加min_random_scale的值:
min_random_scale = 0.623