Exemplo n.º 1
0
def cleanNormal(N_32F, A_8U):
    # N_32F = cv2.bilateralFilter(N_32F, 0, 0.1, 5)
    h, w = N_32F.shape[:2]

    plt.subplot(1, 2, 1)
    plt.gray()
    plt.imshow(normVectors(N_32F.reshape(-1, 3)).reshape(h, w))

    plt.subplot(1, 2, 2)
    plt.gray()
    A_32F = to32F(A_8U)
    A_32F = cv2.GaussianBlur(A_32F, (0, 0), 3.0)
    A_32F = np.clip(10.0 * (A_32F - 0.5) + 0.5, 0.0, 1.0)
    A_32F = cv2.GaussianBlur(A_32F, (0, 0), 3.0)
    N_fix = A_32F > 0.9
    N_bg = A_32F < 0.25
    A_32F = np.clip(10.0 * (A_32F - 0.5) + 0.5, 0.0, 1.0)
    A_8U = to8U(A_32F)
#     plt.imshow(A_8U)
#     plt.show()

    N_32F_blur = cv2.GaussianBlur(N_32F, (0, 0), 3.0)
    for i in xrange(10):
        N_32F_blur = cv2.GaussianBlur(N_32F_blur, (0, 0), 3.0)
        N_32F_blur[N_fix, :] = N_32F[N_fix, :]

    N_32F = N_32F_blur
    # N_32F[N_bg, 2] = 0.0
    N_32F_normalized = normalizeImage(N_32F)

    #A_8U = np.uint8(np.clip(1000.0 * N_32F_normalized[:, :, 2], 0.0, 255.0))
    # A_8U = cv2.bilateralFilter(A_8U, 0, 70, 5)

    return N_32F_normalized, A_8U
Exemplo n.º 2
0
def normalToColor(N_32F, A_8U=None):
    C_32F = 0.5 * N_32F + 0.5
    C_8U = to8U(C_32F)

    if A_8U is not None:
        C_8U = setAlpha(C_8U, A_8U)

    return C_8U
Exemplo n.º 3
0
def isophoteCurves(I_32F, iso_values=[0.2, 0.4, 0.6, 0.8], M_8U=None):
    I_8U = to8U(I_32F)
    isophote_curves = []
    for iso_value in iso_values:
        isophote_curves.append(isophoteCurve(I_8U, iso_value, M_8U))
    return isophote_curves