outlier_correction_mp

sgoldenlab · Nov 8, 2024 · 416e4fb · 416e4fb
1 parent 498d554
commit 416e4fb
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/docs/nb/outlier_correction.ipynb b/docs/nb/outlier_correction.ipynb
@@ -48,8 +48,8 @@
     "\n",
     "## Define if / how you want to interpolate missing pose-estimation data,\n",
     "## and if/how you want to smooth the new pose estimation data: here we do neither.\n",
-    "INTERPOLATION_SETTING = None # OPTIONS: 'None', Animal(s): Nearest', 'Animal(s): Linear', 'Animal(s): Quadratic','Body-parts: Nearest', 'Body-parts: Linear', 'Body-parts: Quadratic'\n",
-    "SMOOTHING_SETTING = None # OPTIONS: {'time_window': 500, 'method': 'savitzky-golay'}, {'time_window': 500, 'method': 'gaussian'}\n",
+    "INTERPOLATION_SETTING = None # EXAMPLE OPTIONS: {'type': 'body-parts', 'method': 'nearest'}, {'type': 'animals', 'method': 'linear'}, {'type': 'animals', 'method': 'quadratic'}, {'type': 'body-parts', 'method': 'linear'}\n",
+    "SMOOTHING_SETTING = None # EXAMPLE OPTIONS: {'time_window': 500, 'method': 'savitzky-golay'}, {'time_window': 500, 'method': 'gaussian'}\n",
     "\n",
     "## Define the fps and the pixels per millimeter of the incoming data: has to be the same for all new videos.\n",
     "## if you have varying fps / px per millimeter / resolutions, then use gui (2023/05)\n",
@@ -137,7 +137,7 @@
   {
    "cell_type": "code",
    "execution_count": 6,
-   "id": "69f8350a",
+   "id": "3067667a",
    "metadata": {},
    "outputs": [
     {
@@ -232,17 +232,17 @@
     "from simba.outlier_tools.outlier_corrector_location_mp import OutlierCorrecterLocationMultiprocess\n",
     "from simba.outlier_tools.outlier_corrector_movement_mp import OutlierCorrecterMovementMultiProcess\n",
     "\n",
-    "CORES = -1 #SPECIFY HOW MANY CPU CORES YOU WANT TO USE. USE -1 \n",
+    "CORE_COUNT = -1 #SPECIFY HOW MANY CPU CORES YOU WANT TO USE. PASS -1 TO USE ALL AVAILABLE CORES. E.g., PASS 8 TO USE 8 CPU CORES.\n",
     "\n",
     "\n",
-    "_ = OutlierCorrecterMovementMultiProcess(config_path=CONFIG_PATH).run()\n",
-    "_ = OutlierCorrecterLocationMultiprocess(config_path=CONFIG_PATH).run()"
+    "_ = OutlierCorrecterMovementMultiProcess(config_path=CONFIG_PATH, core_cnt=CORE_COUNT).run()\n",
+    "_ = OutlierCorrecterLocationMultiprocess(config_path=CONFIG_PATH, core_cnt=CORE_COUNT).run()"
    ]
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e243d26e",
+   "id": "6e24dd44",
    "metadata": {},
    "outputs": [],
    "source": []