Commit cf146f4c authored by David GEISSBUHLER's avatar David GEISSBUHLER

added nan/inf tests for testing

parent eb33d4fd
......@@ -164,10 +164,10 @@ def compute_iqa_specularity_features(rgbImage, startEps=0.05):
speckleFreeImg, diffuseImg, speckleImg = remove_highlights(
rgbImage.astype(np.float32), startEps)
#speckleImg[np.where(np.isinf(speckleImg))] = 0
#speckleImg[np.where(np.isnan(speckleImg))] = 0
#speckleImg[np.where(speckleImg < 0)] = 0
#speckleImg[np.where(speckleImg > 255)] = 255
speckleImg[np.where(np.isinf(speckleImg))] = 0
speckleImg[np.where(np.isnan(speckleImg))] = 0
speckleImg[np.where(speckleImg < 0)] = 0
speckleImg[np.where(speckleImg > 255)] = 255
# speckleImg contains the specular-component
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment