queries.py 文件源码

def get_mask_frame( self, frame, dim=None  ):
        '''
        return mask and name of the query frame. OUTDATED FUNCTION
        '''
        filename = frame.replace(".src.", ".mask.")

        # read image
        ima = cv2.imread(filename)

        # make sure is a mask
        if len(ima.shape)>2:
            ima = ima[:,:,0]

        # binarise
        ima[ima >0]=1.0

        # check dims
        if dim is not None:
            if ima.shape[0]>ima.shape[1]:
                dim_ = dim
            else:
                dim_ = (dim[1], dim[0])
        else:
            dim_ = ima.shape[:2]

        mask_r = reshape_maps_zoom( np.expand_dims(ima, axis=0 ) , dim_).squeeze()
        mask_r[mask_r >0]=1.0

        return mask_r