assert np.all(old_layer[i].data == new_layer[i].data
), 'model architectures do not agree'
else:
assert len(old_layer) == len(new_layer)
for i in range(len(old_layer)):
old_data = old_layer[i].data
new_data = new_layer[i].data
assert new_data.shape == old_data.shape, 'model architectures do not agree'
new_data.flat = old_data.flat
old_layers = list(old_net.params.items())
new_layers = list(new_net.params.items())
for layerno in range(len(old_layers)):
k1, old_layer = old_layers[layerno]
k2, new_layer = new_layers[layerno]
assert len(old_layer) == len(new_layer)
for i in range(len(old_layer)):
assert np.all(old_layer[i].data == new_layer[i].data
), 'model architectures do not agree'
new_net.save(new_weight_path)
if __name__ == '__main__':
r"""
CommandLine:
python -m pysseg.iface_caffe
"""
import ubelt as ub # NOQA
ub.doctest_package()
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import ubelt as ub
ub.doctest_package('futures_actors', 'all')
def main():
import ubelt as ub
ub.doctest_package('wbia.algo.graph', ignore_patterns=['*_grave*'])