def estimateLightDir(input_data):
N_sil = input_data["N_sil"]
I_sil = input_data["I_sil"]
cvs_sil = input_data["cvs_sil"]
I_32F = input_data["I"]
A_8U = input_data["A"]
p_samples = foreGroundSamples(A_8U)
plt.imshow(A_8U)
plt.scatter(p_samples[:, 0], p_samples[:, 1])
plt.show()
N_lumo = lumoNormal(A_8U)
L = estimateLightByCluster(N_lumo, I_32F, p_samples)
output_data = {"L": L}
return output_data
def estimateLightDir(input_data):
N_sil = input_data["N_sil"]
I_sil = input_data["I_sil"]
cvs_sil = input_data["cvs_sil"]
I_32F = input_data["I"]
A_8U = input_data["A"]
N_lumo = lumoNormal(A_8U)
plt.imshow(0.5 * N_lumo + 0.5)
plt.show()
L_phi = estimatePhi(N_sil, I_sil)
output_data = {"L": L_phi}
# L = estimateLbyDistance(cvs_sil, N_sil, I_32F)
# output_data = {"L": L}
L = estimateTheta(L_phi, cvs_sil, N_sil, I_32F, A_8U)
# output_data = {"L": L }
return output_data
