liangyuch commited on
Commit
4981e36
Β·
1 Parent(s): f4cbdf4

debug rel num

Browse files
Files changed (1) hide show
  1. utils.py +2 -1
utils.py CHANGED
@@ -194,7 +194,8 @@ def show_result(img,
194
  # Draw relations
195
 
196
  # Filter out relations
197
- n_rel_topk = num_rel
 
198
  # Exclude background class
199
  rel_dists = result.rel_dists[:, 1:]
200
  # rel_dists = result.rel_dists
 
194
  # Draw relations
195
 
196
  # Filter out relations
197
+ ### Debug: output all relations if not enough
198
+ n_rel_topk = min(num_rel, len(result.labels)//2)
199
  # Exclude background class
200
  rel_dists = result.rel_dists[:, 1:]
201
  # rel_dists = result.rel_dists