Измените некоторые значения в файле веса (.h5) VGG-16. - PullRequest
0 голосов
/ 14 июля 2020

У меня есть значения веса и смещения для каждого слоя модели VGG, сохраненные как файл .h5. Я получил файл по адресу: https://github.com/fchollet/deep-learning-models/releases/tag/v0.1

Теперь предположим, что я хочу изменить несколько значений в этом файле. С помощью Как перезаписать массив внутри файла h5 с помощью h5py , я пытаюсь сделать то же самое следующим образом:

import h5py
file_name = "vgg.h5"

f = h5py.File(file_name, 'r+')

# List all groups
print("Keys: %s" % f.keys())

# Get the data
data = (f['block2_conv1']['block2_conv1_W_1:0'])

print(data[0][0][0][0]) #prints some value, lets say X
data[0][0][0][0] = 0 #change it to zero
print(data[0][0][0][0]) #prints the same value X

f.close()

Я получаю то же исходное значение даже после попытки присвоить что-то еще в этот индекс.

Я не уверен, как изменить / изменить значение веса и сохранить его (в том же файле или, возможно, в другом). Мы будем очень благодарны за любую помощь или предложения по этому поводу. Спасибо!

1 Ответ

0 голосов
/ 19 июля 2020

Попробуйте это.

data[0,0,0,0]=0

Объекты ndarray необходимо обновлять таким образом, а не обычным списком массивов.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...