How to change the input dim of 'train-detector.py'?

[cherishtttz]

Thank u so much for displaying such an amazing project! But I would like to ask if I can change the input 'dim' of script 'train-detector.py' ,because I need to change the shape of input image to rectangle, using '(dimx, dimy)' instead of '(dim, dim)'. If I change the shape, how should I change this function: labels2output_map(label,lppts,dim,stride) in script ./src/sampler.py ?