"} "
"layers { "
" layer { "
+ " name: 'hdf5_output' "
+ " type: 'hdf5_output' "
+ " hdf5_output_param { "
+ " file_name: '/my/hdf5/output/file' "
+ " } "
+ " } "
+ "} "
+ "layers { "
+ " layer { "
" name: 'im2col' "
" type: 'im2col' "
" } "
" type: FLATTEN "
"} "
"layers { "
+ " name: 'hdf5_output' "
+ " type: HDF5_OUTPUT "
+ " hdf5_output_param { "
+ " file_name: '/my/hdf5/output/file' "
+ " } "
+ "} "
+ "layers { "
" name: 'im2col' "
" type: IM2COL "
"} "
is_fully_compatible = false;
}
}
+ if (v0_layer_param.has_hdf5_output_param()) {
+ if (type == "hdf5_output") {
+ layer_param->mutable_hdf5_output_param()->CopyFrom(
+ v0_layer_param.hdf5_output_param());
+ } else {
+ LOG(ERROR) << "Unknown parameter hdf5_output_param for layer type "
+ << type;
+ is_fully_compatible = false;
+ }
+ }
}
return is_fully_compatible;
}
return LayerParameter_LayerType_FLATTEN;
} else if (type == "hdf5_data") {
return LayerParameter_LayerType_HDF5_DATA;
+ } else if (type == "hdf5_output") {
+ return LayerParameter_LayerType_HDF5_OUTPUT;
} else if (type == "im2col") {
return LayerParameter_LayerType_IM2COL;
} else if (type == "images") {