I have the following image which has bee generated using pix2pix model . I want the generated lines to be straight and have the same height (in case of horizontal ones) and same width (in case of vertical ones) .

Basically those lines are walls and should be later on rendered as walls in a house .

Any python /ml logic for that ?