处理头发丝和花之类遮挡问题

libs/MVS/mask_face_occlusion.py

@@ -5,7 +5,7 @@ from scipy.spatial.transform import Rotation
 import sys
 sys.path.append("/home/algo/Documents/mask_face_occlusion/")
 from colmap_loader import read_cameras_text, read_images_text, read_int_text, write_int_text, read_indices_from_file
-from utils.get_pose_matrix import get_w2c
+from get_pose_matrix import get_w2c
 import argparse
 import matplotlib.pyplot as plt
 import collections
@@ -1014,8 +1014,9 @@ class ModelProcessor:
             }
             img_name = img.name[:-4]
             print("img_name=", img_name, n)
-            # if (img_name!="72_2" and img_name!="82_2" and img_name!="83_2"): # 82_2 72_2
-            # if (img_name!="74_8"):
+            # if (img_name!="73_8" and img_name!="52_8" and img_name!="62_8"):
+            # if (img_name!="52_8" and img_name!="62_8"):
+            # if (img_name!="52_8"):
             #    continue
             # face_visibility2, face_contour = self._flag_contour(camera_data, None)
             # countour_faces_dict[img.name[:-4]] = np.where(face_contour)[0].tolist()