updated notebooks

Author: Rabah Abdul Khalek

examples/landmark_detection/criterias/criteria1_partial_faces.ipynb

@@ -19,7 +19,7 @@
     "\n",
     "import torch\n",
     "from giskard_vision.landmark_detection.tests.performance import NMEMean, NMEs, Es, MEMean, MEStd, NMEMean, NMEStd\n",
-    "from giskard_vision.landmark_detection.models.wrappers import FaceAlignmentWrapper"
+    "from giskard_vision.landmark_detection.models.wrappers import OpenCVWrapper "
    ]
   },
   {
@@ -94,7 +94,7 @@
    },
    "outputs": [],
    "source": [
-    "model = FaceAlignmentWrapper(model=FaceAlignment(LandmarksType.TWO_D, device=\"cpu\", flip_input=False))"
+    "model = OpenCVWrapper() #FaceAlignmentWrapper(model=FaceAlignment(LandmarksType.TWO_D, device=\"cpu\", flip_input=False))"
    ]
   },
   {

examples/landmark_detection/features/ex1_draw_landmarks.ipynb

@@ -19,7 +19,7 @@
     "from giskard_vision.landmark_detection.tests.performance import NMEMean\n",
     "from giskard_vision.landmark_detection.tests.base import Test, TestDiff\n",
     "from giskard_vision.landmark_detection.marks.facial_parts import FacialParts\n",
-    "from giskard_vision.landmark_detection.models.wrappers import FaceAlignmentWrapper\n",
+    "from giskard_vision.landmark_detection.models.wrappers import OpenCVWrapper\n",
     "from giskard_vision.landmark_detection.tests.base import Test, TestDiff"
    ]
   },
@@ -95,7 +95,7 @@
    },
    "outputs": [],
    "source": [
-    "model = FaceAlignmentWrapper(model=FaceAlignment(LandmarksType.TWO_D, device=device, flip_input=False))"
+    "model = OpenCVWrapper() # FaceAlignmentWrapper(model=FaceAlignment(LandmarksType.TWO_D, device=device, flip_input=False))"
    ]
   },
   {

examples/landmark_detection/features/ex4_tests_and_metrics.ipynb

@@ -75,7 +75,7 @@
    ],
    "source": [
     "models = {\n",
-    "    \"FaceAlignment\": FaceAlignmentWrapper(model=FaceAlignment(LandmarksType.TWO_D, device=\"cpu\", flip_input=False)),\n",
+    "    \"FaceAlignment\": FaceAlignmentWrapper(model=FaceAlignment(LandmarksType.TWO_D, device=\"cpu\", flip_input=False, face_detector=\"blazeface\")),\n",
     "    \"OpenCV\": OpenCVWrapper(),\n",
     "}\n",
     "# models.pop(\"FaceAlignment\")  # takes a long time"

examples/landmark_detection/report/master.ipynb

@@ -121,7 +121,7 @@
    ],
    "source": [
     "models_list = [\n",
-    "    FaceAlignmentWrapper(model=FaceAlignment(LandmarksType.TWO_D, device=\"cpu\", flip_input=False)),\n",
+    "    FaceAlignmentWrapper(model=FaceAlignment(LandmarksType.TWO_D, device=\"cpu\", flip_input=False, face_detector=\"blazeface\")),\n",
     "    OpenCVWrapper(),\n",
     "]\n",
     "\n",