Add centerface-pytorch.ipynb

centerface-pytorch.ipynb

@@ -120,7 +120,7 @@
    "outputs": [],
    "source": [
     "# 执行以下代码，将预置标注数据转换成coco形式\n",
-    "!python ./datasets/gen_data.py"
+    "!python3 datasets/gen_data.py"
    ]
   },
   {
@@ -184,7 +184,7 @@
    "outputs": [],
    "source": [
     "# 单卡训练\n",
-    "!bash ./src/train.sh"
+    "!python3 src/main.py --input_h 640 --input_w 640 --batch_size 8 --lr 5e-4 --data_dir datasets/"
    ]
   },
   {
@@ -195,7 +195,7 @@
    "outputs": [],
    "source": [
     "# 多卡训练（需要2个加速卡）\n",
-    "!bash ./src/train_multi.sh"
+    "!python3 src/main.py --input_h 640 --input_w 640 --batch_size 8 --lr 5e-4 --data_dir datasets/ --gpus 0,1"
    ]
   },
   {
@@ -216,10 +216,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!cd src/lib/external/\n",
-    "!bash make.sh\n",
-    "!cd ../../\n",
-    "!python3 test_wider_face.py\n",
+    "!bash src/lib/external/make.sh\n",
+    "!python3 src/test_wider_face.py\n",
     "# 执行完成后会在当前目录下生成draw_img.jpg的图像"
    ]
   },