-
Notifications
You must be signed in to change notification settings - Fork 19
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
teng
committed
Mar 15, 2021
1 parent
485436d
commit 6c91a3d
Showing
175 changed files
with
92,537 additions
and
38 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,103 @@ | ||
| // | ||
| // NanoDet.h | ||
| // TNNDemo | ||
| // | ||
| // Created by WZTENG on 2021/1/11. | ||
| // Copyright © 2021 TENG. All rights reserved. | ||
| // | ||
|
|
||
| #ifndef NANODET_H | ||
| #define NANODET_H | ||
|
|
||
| #include <stdio.h> | ||
| #import <Foundation/Foundation.h> | ||
| #import <UIKit/UIKit.h> | ||
| #import <UIKit/UIImage.h> | ||
| #import <functional> | ||
|
|
||
| #include "tnn/core/tnn.h" | ||
| #include "tnn/core/blob.h" | ||
| #include "tnn/utils/blob_converter.h" | ||
| #include "tnn/utils/mat_utils.h" | ||
|
|
||
| #include "UIImage+Utility.h" | ||
| #include <Metal/Metal.h> | ||
|
|
||
| #include "Yolov5.h" | ||
|
|
||
| typedef struct { | ||
| std::string cls_layer; | ||
| std::string dis_layer; | ||
| int stride; | ||
| } HeadInfo; | ||
|
|
||
| //typedef struct { | ||
| // float x1; | ||
| // float y1; | ||
| // float x2; | ||
| // float y2; | ||
| // float score; | ||
| // int label; | ||
| //} BoxInfo; | ||
|
|
||
| class NanoDet { | ||
| public: | ||
| NanoDet(bool useGPU); | ||
|
|
||
| ~NanoDet(); | ||
|
|
||
| std::vector<BoxInfo> detect(UIImage *image, float threshold, float nms_threshold); | ||
|
|
||
| std::vector<std::string> labels{"person", "bicycle", "car", "motorcycle", "airplane", "bus", "train", "truck", "boat", "traffic light", | ||
| "fire hydrant", "stop sign", "parking meter", "bench", "bird", "cat", "dog", "horse", "sheep", "cow", | ||
| "elephant", "bear", "zebra", "giraffe", "backpack", "umbrella", "handbag", "tie", "suitcase", "frisbee", | ||
| "skis", "snowboard", "sports ball", "kite", "baseball bat", "baseball glove", "skateboard", "surfboard", | ||
| "tennis racket", "bottle", "wine glass", "cup", "fork", "knife", "spoon", "bowl", "banana", "apple", | ||
| "sandwich", "orange", "broccoli", "carrot", "hot dog", "pizza", "donut", "cake", "chair", "couch", | ||
| "potted plant", "bed", "dining table", "toilet", "tv", "laptop", "mouse", "remote", "keyboard", "cell phone", | ||
| "microwave", "oven", "toaster", "sink", "refrigerator", "book", "clock", "vase", "scissors", "teddy bear", | ||
| "hair drier", "toothbrush"}; | ||
|
|
||
| private: | ||
| void decode_infer(const std::shared_ptr<TNN_NS::Mat>& cls_pred, | ||
| const std::shared_ptr<TNN_NS::Mat>& dis_pred, | ||
| int stride, float threshold, std::vector<std::vector<BoxInfo>> &results); | ||
|
|
||
| BoxInfo disPred2Bbox(const float *&dfl_det, int label, float score, int x, int y, int stride); | ||
|
|
||
| void nms(std::vector<BoxInfo> &input_boxes, float NMS_THRESH); | ||
|
|
||
| private: | ||
| std::shared_ptr<tnn::TNN> net; | ||
| std::shared_ptr<TNN_NS::Instance> instance; | ||
| TNN_NS::DeviceType device_type; | ||
|
|
||
| int in_n = 1; | ||
| int in_c = 3; | ||
| int in_w = 320; | ||
| int in_h = 320; | ||
|
|
||
| const int num_class = 80; | ||
| const int reg_max = 7; | ||
|
|
||
| std::vector<HeadInfo> heads_info{ | ||
| // cls_pred | dis_pred | stride | ||
| {"792", "795", 8}, | ||
| {"814", "817", 16}, | ||
| {"836", "839", 32}, | ||
| }; | ||
|
|
||
| public: | ||
| static NanoDet *detector; | ||
| static bool hasGPU; | ||
| static bool toUseGPU; | ||
| }; | ||
|
|
||
| template<typename _Tp> | ||
| int activation_function_softmax(const _Tp *src, _Tp *dst, int length); | ||
|
|
||
| inline float fast_exp(float x); | ||
|
|
||
| inline float sigmoid(float x); | ||
|
|
||
| #endif //NANODET_H |
Oops, something went wrong.