IVALab Python Libraries
Collection of code for computer vision and robotics with specific API.
Namespaces | Functions
imageProcessing.py File Reference

Namespaces

 puzzle.utils.imageProcessing
 

Functions

def cropImage (image, template)
 Crop and resize a cover image, which has the same shape with the mask. More...
 
def extract_region (img, verbose=False)
 
def find_nonzero_mask (mask)
 Extract the coordinates of the non-zero elements (x,y style) from a mask image. More...
 
def preprocess_real_puzzle (img, mask=None, areaThresholdLower=1000, areaThresholdUpper=8000, BoudingboxThresh=(30, 80), cannyThresh=(30, 60), WITH_AREA_THRESH=False, verbose=False, ret_thresh_mask=False)
 Preprocess the RGB image of a segmented puzzle piece in a circle area to obtain a mask. More...
 
def preprocess_synthetic_puzzle (img, mask=None, areaThresholdLower=1000, areaThresholdUpper=8000, cannyThresh=(20, 80), verbose=False)
 Preprocess the RGB image of a segmented puzzle piece in a circle area to obtain a mask. More...
 
def rotate_im (image, angle, mask=None)
 Generate rotated image and return working elements of it. More...
 
def rotate_nd (image, angle, mask=None)
 Generate rotated image and return working elements of it. More...
 
def white_balance (img)