Ge Zheng commited on
Commit
dbd24dd
Β·
1 Parent(s): a6a308d

chore(demo): update megengine deployment (#153)

Browse files
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.65
19
- #define BBOX_CONF_THRESH 0.3
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.3 --nms 0.65 --tsize 640 --save_result
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