Skip to content

Commit

Permalink
[fix doc path error]
Browse files Browse the repository at this point in the history
  • Loading branch information
zen committed Aug 11, 2023
1 parent e8b1436 commit 707ba98
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion RKOPT_README_zh.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@

- 假设有 6000 个候选框,检测类别是 80 类,则阈值检索操作需要重复 6000* 80 ~= 4.8*10^5 次,占据了较多耗时。故导出模型时,在模型中额外新增了对 80 类检测目标进行求和操作,用于快速过滤置信度。(该结构在部分情况下对有效,与模型的训练结果有关)

可以在 **./ultralytics/nn/modules.py** 52行~54行的位置,注释掉这部分优化,对应的代码是:
可以在 **./ultralytics/nn/modules/head.py** 52行~54行的位置,注释掉这部分优化,对应的代码是:

```
cls_sum = torch.clamp(y[-1].sum(1, keepdim=True), 0, 1)
Expand Down

0 comments on commit 707ba98

Please sign in to comment.