a big tree refine

idvhridoy · Jun 19, 2021 · ce3600a · ce3600a
1 parent 5ca3f99
commit ce3600a
Show file tree

Hide file tree

Showing 167 changed files with 3,747 additions and 16,144 deletions.
diff --git a/.gitignore b/.gitignore
@@ -99,3 +99,5 @@ ENV/
 
 # mypy
 .mypy_cache/
+
+.DS_Store
diff --git a/README.md b/README.md
diff --git a/alignment/README.md b/alignment/README.md
@@ -1,4 +1,42 @@
-You can now find heatmap based approaches under ``heatmapReg`` directory.
+## Face Alignment
+
+
+<div align="left">
+  <img src="https://insightface.ai/assets/img/custom/logo3.jpg" width="240"/>
+</div>
+
+
+## Introduction
+
+These are the face alignment methods of [InsightFace](https://insightface.ai)
+
+
+<div align="left">
+  <img src="https://insightface.ai/assets/img/custom/thumb_sdunet.png" width="600"/>
+</div>
+
+
+### Datasets
+
+  Please refer to [datasets](_datasets_) page for the details of face alignment datasets used for training and evaluation.
+
+### Evaluation
+
+  Please refer to [evaluation](_evaluation_) page for the details of face alignment evaluation.
+
+
+## Methods
+
+
+Supported methods:
+
+- [x] [SDUNets (BMVC'2018)](heatmap)
+- [x] [SimpleRegression](coordinate_reg)
+
+
+
+## Contributing
+
+We appreciate all contributions to improve the face alignment model zoo of InsightFace. 
 
-You can now find coordinate regression approaches under ``coordinateReg`` directory.
 
diff --git a/alignment/coordinateReg/README.md → alignment/coordinate_reg/README.md b/alignment/coordinateReg/README.md → alignment/coordinate_reg/README.md
diff --git a/alignment/coordinateReg/image_infer.py → alignment/coordinate_reg/image_infer.py b/alignment/coordinateReg/image_infer.py → alignment/coordinate_reg/image_infer.py
diff --git a/alignment/heatmapReg/README.md → alignment/heatmap/README.md b/alignment/heatmapReg/README.md → alignment/heatmap/README.md
diff --git a/alignment/heatmapReg/data.py → alignment/heatmap/data.py b/alignment/heatmapReg/data.py → alignment/heatmap/data.py
diff --git a/alignment/heatmapReg/img_helper.py → alignment/heatmap/img_helper.py b/alignment/heatmapReg/img_helper.py → alignment/heatmap/img_helper.py
diff --git a/alignment/heatmapReg/metric.py → alignment/heatmap/metric.py b/alignment/heatmapReg/metric.py → alignment/heatmap/metric.py
diff --git a/alignment/heatmapReg/optimizer.py → alignment/heatmap/optimizer.py b/alignment/heatmapReg/optimizer.py → alignment/heatmap/optimizer.py
diff --git a/alignment/heatmapReg/sample_config.py → alignment/heatmap/sample_config.py b/alignment/heatmapReg/sample_config.py → alignment/heatmap/sample_config.py
diff --git a/alignment/heatmapReg/symbol/sym_heatmap.py → alignment/heatmap/symbol/sym_heatmap.py b/alignment/heatmapReg/symbol/sym_heatmap.py → alignment/heatmap/symbol/sym_heatmap.py
diff --git a/alignment/heatmapReg/test.py → alignment/heatmap/test.py b/alignment/heatmapReg/test.py → alignment/heatmap/test.py
diff --git a/alignment/heatmapReg/test_rec_nme.py → alignment/heatmap/test_rec_nme.py b/alignment/heatmapReg/test_rec_nme.py → alignment/heatmap/test_rec_nme.py
diff --git a/alignment/heatmapReg/train.py → alignment/heatmap/train.py b/alignment/heatmapReg/train.py → alignment/heatmap/train.py
diff --git a/attribute/README.md b/attribute/README.md
@@ -0,0 +1,41 @@
+## Face Alignment
+
+
+<div align="left">
+  <img src="https://insightface.ai/assets/img/custom/logo3.jpg" width="320"/>
+</div>
+
+
+## Introduction
+
+These are the face attribute methods of [InsightFace](https://insightface.ai)
+
+
+<div align="left">
+  <img src="https://insightface.ai/assets/img/github/t1_genderage.jpg" width="600"/>
+</div>
+
+
+### Datasets
+
+  Please refer to [datasets](_datasets_) page for the details of face attribute datasets used for training and evaluation.
+
+### Evaluation
+
+  Please refer to [evaluation](_evaluation_) page for the details of face attribute evaluation.
+
+
+## Methods
+
+
+Supported methods:
+
+- [x] [Gender_Age](gender_age)
+
+
+
+## Contributing
+
+We appreciate all contributions to improve the face attribute model zoo of InsightFace. 
+
+
diff --git a/deploy/test.py → attribute/gender_age/test.py b/deploy/test.py → attribute/gender_age/test.py
@@ -4,25 +4,21 @@
 import numpy as np
 import insightface
 from insightface.app import FaceAnalysis
+from insightface.data import get_image as ins_get_image
 
-assert insightface.__version__>='0.2'
 
-parser = argparse.ArgumentParser(description='insightface test')
+parser = argparse.ArgumentParser(description='insightface gender-age test')
 # general
 parser.add_argument('--ctx', default=0, type=int, help='ctx id, <0 means using cpu')
 args = parser.parse_args()
 
-app = FaceAnalysis(name='antelope')
+app = FaceAnalysis(allowed_modules=['detection', 'genderage'])
 app.prepare(ctx_id=args.ctx, det_size=(640,640))
 
-img = cv2.imread('../sample-images/t1.jpg')
+img = ins_get_image('t1')
 faces = app.get(img)
 assert len(faces)==6
-rimg = app.draw_on(img, faces)
-cv2.imwrite("./t1_output.jpg", rimg)
-print(len(faces))
 for face in faces:
     print(face.bbox)
-    print(face.kps)
-    print(face.embedding.shape)
+    print(face.sex, face.age)
 
diff --git a/challenges/README.md b/challenges/README.md
@@ -0,0 +1,31 @@
+## Challenges
+
+
+<div align="left">
+  <img src="https://insightface.ai/assets/img/custom/logo3.jpg" width="240"/>
+</div>
+
+
+## Introduction
+
+These are challenges hold by [InsightFace](https://insightface.ai)
+
+
+<div align="left">
+  <img src="https://insightface.ai/assets/img/custom/thumb_ifrt.png" width="480"/>
+</div>
+
+
+
+## List
+
+
+Supported methods:
+
+- [LFR19 (ICCVW'2019)](iccv19-lfr)
+- [MFR21 (ICCVW'2021)](iccv21-mfr)
+- [IFRT](ifrt)
+
+
+
+
diff --git a/challenges/iccv19-lfr/README.md b/challenges/iccv19-lfr/README.md
@@ -31,7 +31,7 @@ [email protected]
 
 *For Chinese:*
 
-![wechat](https://github.com/deepinsight/insightface/blob/master/resources/lfr19_wechat1.jpg)
+![wechat](https://insightface.ai/assets/img/github/lfr19_wechat1.jpg)
 
 *For English:*
 

diff --git a/deploy/README.md b/deploy/README.md
diff --git a/deploy/convert_onnx.py b/deploy/convert_onnx.py
diff --git a/deploy/face_model.py b/deploy/face_model.py
diff --git a/deploy/model_slim.py b/deploy/model_slim.py
diff --git a/deploy/mtcnn-model/det1-0001.params b/deploy/mtcnn-model/det1-0001.params