Skip to content

Commit

Permalink
feat(toolkit): 🍅 add SCRFD MNN/NCNN/TNN/ONNXRuntime
Browse files Browse the repository at this point in the history
  • Loading branch information
DefTruth committed Jan 16, 2022
1 parent daf8439 commit 8bf108f
Show file tree
Hide file tree
Showing 14 changed files with 57 additions and 0 deletions.
1 change: 1 addition & 0 deletions examples/lite/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -73,4 +73,5 @@ add_lite_executable(lite_yolov5_v6.0 cv)
add_lite_executable(lite_mg_matting cv)
add_lite_executable(lite_nanodet_plus cv)
add_lite_executable(lite_scrfd cv)
add_lite_executable(lite_yolo5face cv)

4 changes: 4 additions & 0 deletions examples/lite/cv/test_lite_yolo5face.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
//
// Created by DefTruth on 2022/1/16.
//

1 change: 1 addition & 0 deletions lite/mnn/core/mnn_core.h
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,7 @@ namespace mnncv
class LITE_EXPORTS MNNMGMatting; // [55] * reference: https://github.com/yucornetto/MGMatting
class LITE_EXPORTS MNNNanoDetPlus; // [56] * reference: https://github.com/RangiLyu/nanodet
class LITE_EXPORTS MNNSCRFD; // [57] * reference: https://github.com/deepinsight/insightface/tree/master/detection/scrfd
class LITE_EXPORTS MNNYOLO5Face; // [58] * reference: https://github.com/deepcam-cn/yolov5-face

}

Expand Down
4 changes: 4 additions & 0 deletions lite/mnn/cv/mnn_yolo5face.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
//
// Created by DefTruth on 2022/1/16.
//

8 changes: 8 additions & 0 deletions lite/mnn/cv/mnn_yolo5face.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
//
// Created by DefTruth on 2022/1/16.
//

#ifndef LITE_AI_TOOLKIT_MNN_CV_MNN_YOLO5FACE_H
#define LITE_AI_TOOLKIT_MNN_CV_MNN_YOLO5FACE_H

#endif //LITE_AI_TOOLKIT_MNN_CV_MNN_YOLO5FACE_H
1 change: 1 addition & 0 deletions lite/ncnn/core/ncnn_core.h
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,7 @@ namespace ncnncv
class LITE_EXPORTS NCNNFCNResNet101; // [55] * reference: https://pytorch.org/hub/pytorch_vision_fcn_resnet101/
class LITE_EXPORTS NCNNNanoDetPlus; // [56] * reference: https://github.com/RangiLyu/nanodet
class LITE_EXPORTS NCNNSCRFD; // [57] * reference: https://github.com/deepinsight/insightface/tree/master/detection/scrfd
class LITE_EXPORTS NCNNYOLO5Face; // [58] * reference: https://github.com/deepcam-cn/yolov5-face
}

namespace ncnncv
Expand Down
4 changes: 4 additions & 0 deletions lite/ncnn/cv/ncnn_yolo5face.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
//
// Created by DefTruth on 2022/1/16.
//

8 changes: 8 additions & 0 deletions lite/ncnn/cv/ncnn_yolo5face.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
//
// Created by DefTruth on 2022/1/16.
//

#ifndef LITE_AI_TOOLKIT_NCNN_CV_NCNN_YOLO5FACE_H
#define LITE_AI_TOOLKIT_NCNN_CV_NCNN_YOLO5FACE_H

#endif //LITE_AI_TOOLKIT_NCNN_CV_NCNN_YOLO5FACE_H
1 change: 1 addition & 0 deletions lite/ort/core/ort_core.h
Original file line number Diff line number Diff line change
Expand Up @@ -95,6 +95,7 @@ namespace ortcv
class LITE_EXPORTS MGMatting; // [80] * reference: https://github.com/yucornetto/MGMatting
class LITE_EXPORTS NanoDetPlus; // [81] * reference: https://github.com/RangiLyu/nanodet
class LITE_EXPORTS SCRFD; // [82] * reference: https://github.com/deepinsight/insightface/tree/master/detection/scrfd
class LITE_EXPORTS YOLO5Face; // [83] * reference: https://github.com/deepcam-cn/yolov5-face

}

Expand Down
4 changes: 4 additions & 0 deletions lite/ort/cv/yolo5face.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
//
// Created by DefTruth on 2022/1/16.
//

8 changes: 8 additions & 0 deletions lite/ort/cv/yolo5face.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
//
// Created by DefTruth on 2022/1/16.
//

#ifndef LITE_AI_TOOLKIT_ORT_CV_YOLO5FACE_H
#define LITE_AI_TOOLKIT_ORT_CV_YOLO5FACE_H

#endif //LITE_AI_TOOLKIT_ORT_CV_YOLO5FACE_H
1 change: 1 addition & 0 deletions lite/tnn/core/tnn_core.h
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,7 @@ namespace tnncv
class LITE_EXPORTS TNNMGMatting; // [55] * reference: https://github.com/yucornetto/MGMatting
class LITE_EXPORTS TNNNanoDetPlus; // [56] * reference: https://github.com/RangiLyu/nanodet
class LITE_EXPORTS TNNSCRFD; // [57] * reference: https://github.com/deepinsight/insightface/tree/master/detection/scrfd
class LITE_EXPORTS TNNYOLO5Face; // [58] * reference: https://github.com/deepcam-cn/yolov5-face

}

Expand Down
4 changes: 4 additions & 0 deletions lite/tnn/cv/tnn_yolo5face.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
//
// Created by DefTruth on 2022/1/16.
//

8 changes: 8 additions & 0 deletions lite/tnn/cv/tnn_yolo5face.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
//
// Created by DefTruth on 2022/1/16.
//

#ifndef LITE_AI_TOOLKIT_TNN_CV_TNN_YOLO5FACE_H
#define LITE_AI_TOOLKIT_TNN_CV_TNN_YOLO5FACE_H

#endif //LITE_AI_TOOLKIT_TNN_CV_TNN_YOLO5FACE_H

0 comments on commit 8bf108f

Please sign in to comment.