Ge Zheng
commited on
Commit
Β·
dbd24dd
1
Parent(s):
a6a308d
chore(demo): update megengine deployment (#153)
Browse files- README.md +1 -0
- demo/MegEngine/cpp/yolox.cpp +2 -2
- demo/MegEngine/python/README.md +1 -1
README.md
CHANGED
@@ -8,6 +8,7 @@ For more details, please refer to our [report on Arxiv](https://arxiv.org/abs/21
|
|
8 |
<img src="assets/git_fig.png" width="1000" >
|
9 |
|
10 |
## Updates!!
|
|
|
11 |
* γ2021/07/20γ We have released our technical report on [Arxiv](https://arxiv.org/abs/2107.08430).
|
12 |
|
13 |
## Comming soon
|
|
|
8 |
<img src="assets/git_fig.png" width="1000" >
|
9 |
|
10 |
## Updates!!
|
11 |
+
* γ2021/07/26γ We now support [MegEngine](https://github.com/Megvii-BaseDetection/YOLOX/tree/main/demo/MegEngine) deployment.
|
12 |
* γ2021/07/20γ We have released our technical report on [Arxiv](https://arxiv.org/abs/2107.08430).
|
13 |
|
14 |
## Comming soon
|
demo/MegEngine/cpp/yolox.cpp
CHANGED
@@ -15,8 +15,8 @@
|
|
15 |
/**
|
16 |
* @brief Define names based depends on Unicode path support
|
17 |
*/
|
18 |
-
#define NMS_THRESH 0.
|
19 |
-
#define BBOX_CONF_THRESH 0.
|
20 |
|
21 |
constexpr int INPUT_W = 640;
|
22 |
constexpr int INPUT_H = 640;
|
|
|
15 |
/**
|
16 |
* @brief Define names based depends on Unicode path support
|
17 |
*/
|
18 |
+
#define NMS_THRESH 0.45
|
19 |
+
#define BBOX_CONF_THRESH 0.25
|
20 |
|
21 |
constexpr int INPUT_W = 640;
|
22 |
constexpr int INPUT_H = 640;
|
demo/MegEngine/python/README.md
CHANGED
@@ -21,7 +21,7 @@ python3 convert_weights.py -w yolox_s.pth.tar -o yolox_s_mge.pkl
|
|
21 |
This part is the same as torch's python demo, but no need to specify device.
|
22 |
|
23 |
```
|
24 |
-
python3 demo.py image -n yolox-s -c yolox_s_mge.pkl --path ../../../assets/dog.jpg --conf 0.
|
25 |
```
|
26 |
|
27 |
### [Optional]Step4: dump model for cpp inference
|
|
|
21 |
This part is the same as torch's python demo, but no need to specify device.
|
22 |
|
23 |
```
|
24 |
+
python3 demo.py image -n yolox-s -c yolox_s_mge.pkl --path ../../../assets/dog.jpg --conf 0.25 --nms 0.45 --tsize 640 --save_result
|
25 |
```
|
26 |
|
27 |
### [Optional]Step4: dump model for cpp inference
|