How to run the code

juppiiiii · Feb 21, 2021 · c1c8428 · c1c8428
1 parent ccc5be7
commit c1c8428
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -1 +1,18 @@
-# inference-using-detectron2
+# inference-using-detectron2
+
+#### Start !
+```shell
+python inference.py --mode directory --dir {dir_name} --save {save_dir_name} \
+        --model COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml
+```
+
+
+
+#### Single image inference
+```shell
+python inference.py --mode single-img --single-imge ./test.png --save {save_dir_name} \
+        --model COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml
+```
+
+###### reference  
+https://github.com/facebookresearch/detectron2