1. Z is built by making average on RoIs of the same class, how to deal with the different size of RoIs in one class? 2. How can I get the dict Z of my customized dataset?