I wanted to implement DRAEM_rect, whose mask shape is rectangle. Can you please provide code which can generate the random rectangle masks? Thanks!!