Я выполняю ML-examples ->cmsisnn-cifar10
. Успешно сгенерировать средний файл mean.binaryproto,ancifar10_test_lmdb and cifar10_train_lmdb
. При преобразовании обученной модели Caffe в файл .pkl
, чтобы ее можно было преобразовать в исходный файл cmsis-nn
с помощью следующей команды.
python nn_quantizer.py --model models/cifar10_m7_train_test.prototxt --weights models/cifar10_m7_iter_300000.caffemodel.h5 --save models/cifar10_m7.pkl
получил ошибку ниже
Check failed: blob_dims == blob->shape() Cannot load blob from hdf5;
shape mismatch. Source shape is 16 32 5 5 (12800) target shape is 32 32 5 5 (25600)
Помогите мнеЯ новичок в Кафе.