computation with the extended parts model works better now
get_part_features ensures that the extended parts have the same length as the input edge map so that we aren't decreasing time length anywhere
in get_params the computation of the number of channels for the extended parts works