From 376f0b075bcdf4c460b7db72a2a40207b8650fb5 Mon Sep 17 00:00:00 2001
From: "Kin-Yiu, Wong" <102582011@cc.ncu.edu.tw>
Date: Thu, 11 Aug 2022 19:17:24 +0800
Subject: [PATCH] main code
update readme
---
README.md | 44 +++++++++++++++++++++++++-------------------
1 file changed, 25 insertions(+), 19 deletions(-)
diff --git a/README.md b/README.md
index 6242ee1e..5834eb4d 100644
--- a/README.md
+++ b/README.md
@@ -128,12 +128,6 @@ python train_aux.py --workers 8 --device 0 --batch-size 16 --data data/custom.ya
See [reparameterization.ipynb](tools/reparameterization.ipynb)
-## Pose estimation
-
-[`yolov7-w6-pose.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-w6-pose.pt)
-
-See [keypoint.ipynb](https://github.com/WongKinYiu/yolov7/blob/main/tools/keypoint.ipynb).
-
## Inference
On video:
@@ -188,6 +182,31 @@ python ./tensorrt-python/export.py -o yolov7-tiny.onnx -e yolov7-tiny-nms.trt -p
Tested with: Python 3.7.13, Pytorch 1.12.0+cu113
+## Pose estimation
+
+[`code`](https://github.com/WongKinYiu/yolov7/tree/pose) [`yolov7-w6-pose.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-w6-pose.pt)
+
+See [keypoint.ipynb](https://github.com/WongKinYiu/yolov7/blob/main/tools/keypoint.ipynb).
+
+
+
+
+## Instance segmentation
+
+[`code`](https://github.com/WongKinYiu/yolov7/tree/mask) [`yolov7-mask.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-mask.pt)
+
+See [instance.ipynb](https://github.com/WongKinYiu/yolov7/blob/main/tools/instance.ipynb).
+
+
+
## Citation
@@ -200,19 +219,6 @@ Tested with: Python 3.7.13, Pytorch 1.12.0+cu113
}
```
-## Teaser
-
-Yolov7-mask & YOLOv7-pose
-
-
-
## Acknowledgements
+
+
+
+