DogBreed_gluon icon indicating copy to clipboard operation
DogBreed_gluon copied to clipboard

features shape

Open thormacy opened this issue 6 years ago • 2 comments

I am trying to run the model, but there's a error in this line: features = np.concatenate([features_resnet, features_densenet, features_inception], axis=-1) Error is caused by the inconsistent dimensions.

Then I check the shape of features_resnet, features_densenet, features_inception. Find that their shapes are not the same. I pretrain the features using the preprocessing file.

thormacy avatar Mar 31 '18 19:03 thormacy

Thanks for any suggestion

thormacy avatar Mar 31 '18 19:03 thormacy

可能由于 gluon 的更新,你需要加一个 GlobalAveragePooling。

ypwhs avatar Apr 01 '18 07:04 ypwhs