apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

main code

Browse Source 
set top_k temporally
This commit is contained in:
Kin-Yiu, Wong 2022-07-08 17:12:36 +08:00 committed by GitHub
parent 8f2e9e701c
commit 5f1b78ad61
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/loss.py
View File

@ -712,7 +712,7 @@ class ComputeLossOTA:
pair_wise_iou_loss = -torch.log(pair_wise_iou + 1e-8) pair_wise_iou_loss = -torch.log(pair_wise_iou + 1e-8)
top_k, _ = torch.topk(pair_wise_iou, min(20, pair_wise_iou.shape[1]), dim=1) top_k, _ = torch.topk(pair_wise_iou, min(10, pair_wise_iou.shape[1]), dim=1)
dynamic_ks = torch.clamp(top_k.sum(1).int(), min=1) dynamic_ks = torch.clamp(top_k.sum(1).int(), min=1)
gt_cls_per_image = ( gt_cls_per_image = (