Update colmap2nerf.py

handle reflection case with small noise

scripts/colmap2nerf.py

@@ -128,11 +128,11 @@ def qvec2rotmat(qvec):
 
 def rotmat(a, b):
 	a, b = a / np.linalg.norm(a), b / np.linalg.norm(b)
+	v = np.cross(a, b)
 	c = np.dot(a, b)
 	# handle exception for the opposite direction input
-	if(c < -1 + 1e-10):
-		return -np.eye(3)
-	v = np.cross(a, b)
+	if c < -1 + 1e-10:
+		return rotmat(a + np.random.uniform(-1e-2, 1e-2, 3), b)
 	s = np.linalg.norm(v)
 	kmat = np.array([[0, -v[2], v[1]], [v[2], 0, -v[0]], [-v[1], v[0], 0]])
 	return np.eye(3) + kmat + kmat.dot(kmat) * ((1 - c) / (s ** 2 + 1e-10))