rectifyAffine.py

#!/anaconda/bin/python
import matplotlib.image as mplimg
import os
import sys
from utils import *

# Apply affine rectification to an image using pairs of parallel lines

# ---------- Settings ---------
nLinePairs = 2; # Select this may pairs of perpendicular lines
# -----------------------------
def usage(sname):
    print(sname + " <image> <# line pairs (>1)>");

if len(sys.argv) != 2 and len(sys.argv) != 3:
    usage(sys.argv[0])
    exit(0)

if len(sys.argv) == 3:
    nLinePairs = int(sys.argv[2])
    if nLinePairs < 2:
        usage(sys.argv[0])

print(f"Affine rectification via the vanishing line using {d} parallel line pairs {nLinePairs}" )

# Read input files
imgPath = sys.argv[1];
fileparts = os.path.splitext(imgPath)
if fileparts[1] == '':
    fileparts = (fileparts[0], '.png')
imgPath = fileparts[0] + fileparts[1];
im = mplimg.imread(imgPath)

# Do affine rectification
imA, HA = rectifyAffineF(im, nLinePairs)
plt.close('all')
    
# Show output
imshow(np.concatenate((im,imA),axis=1))
axis('off')
plt.suptitle('Original (left), Affine rectified (right)')
margin = 0.05;
plt.subplots_adjust(left=margin, right=1.0-margin, top=1.0-margin, bottom=margin)
show()

# Save output
imgPathRect = fileparts[0] + "_arect" + fileparts[1];
print(("Saving output to %s..." % imgPathRect))
mplimg.imsave(imgPathRect,cropOuterRegion(imA))