Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -264,7 +264,7 @@ def main():
|
|
| 264 |
if args.binary:
|
| 265 |
suitable_count = results['suitable_for_vqa'].sum()
|
| 266 |
else:
|
| 267 |
-
suitable_count = (results['prediction'] > 0.
|
| 268 |
|
| 269 |
total_count = len(results)
|
| 270 |
|
|
|
|
| 264 |
if args.binary:
|
| 265 |
suitable_count = results['suitable_for_vqa'].sum()
|
| 266 |
else:
|
| 267 |
+
suitable_count = (results['prediction'] > 0.9).sum()
|
| 268 |
|
| 269 |
total_count = len(results)
|
| 270 |
|