Spaces:
Build error
Build error
debug rel num
Browse files
utils.py
CHANGED
@@ -194,7 +194,8 @@ def show_result(img,
|
|
194 |
# Draw relations
|
195 |
|
196 |
# Filter out relations
|
197 |
-
|
|
|
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
|