number_sequence fix base layout
keep square layout to avoid distorting images. TODO: fix all images and dataset to use 520*520 instead of 800*520
(cherry picked from commit 2206ece7ac060909906edf44bb757ef1633cc5b7)