Как узнать количество слоев из сохраненного файла h5 предварительно обученной модели?

Question

У меня сохранены веса VGG-16 в формате h5.Я хочу видеть количество слоев в сети.Как я могу это сделать?

Я пытался использовать:

file = h5py.File(vgg16.h5)

, после этого я проверил на file.attrs, но после этого я не знаю, какую команду использовать, чтобы найтиколичество слоев в сети.

Answer 1

Без ваших данных или кода трудно предоставить более подробную информацию.Чтобы продемонстрировать методы h5py для доступа к данным h5, вот простой пример, который создает файл h5 с 1 группой из 3 наборов данных.После того, как группа и наборы данных созданы, существует цикл для печати имени, формы и типа набора данных.

import h5py, numpy as np
h5f=h5py.File('SO_54511719.h5','w')

ds_data = np.random.random(100).reshape(10,10)
group1 = h5f.create_group('group1')
group1.create_dataset('ds_1', data=ds_data)
group1.create_dataset('ds_2', data=ds_data)
group1.create_dataset('ds_3', data=ds_data)

print ('number of datasets in group:', len(group1))
for (dsname, dsvalue) in group1.items() :
    print ('for',dsname,':')
    print ('shape:',dsvalue.shape)
    print ('dtype:',dsvalue.dtype)

h5f.close()

Вывод выглядит следующим образом:

number of datasets in group: 3
for ds_1 :
shape: (10, 10)
dtype: float64
for ds_2 :
shape: (10, 10)
dtype: float64
for ds_3 :
shape: (10, 10)
dtype: float64