visualization update
Browse files- infer.py +1 -1
- requirements.txt +1 -0
infer.py
CHANGED
|
@@ -16,7 +16,7 @@ class detections():
|
|
| 16 |
self.all_preds = []
|
| 17 |
self.all_scores = []
|
| 18 |
self.index = []
|
| 19 |
-
self.score_thr = 0.
|
| 20 |
self.result = []
|
| 21 |
self.record_dict = {'model': cfg_path,'results': []}
|
| 22 |
self.detect_count = []
|
|
|
|
| 16 |
self.all_preds = []
|
| 17 |
self.all_scores = []
|
| 18 |
self.index = []
|
| 19 |
+
self.score_thr = 0.9
|
| 20 |
self.result = []
|
| 21 |
self.record_dict = {'model': cfg_path,'results': []}
|
| 22 |
self.detect_count = []
|
requirements.txt
CHANGED
|
@@ -8,3 +8,4 @@ torchvision==0.10.0
|
|
| 8 |
imantics
|
| 9 |
terminaltables
|
| 10 |
pycocotools
|
|
|
|
|
|
| 8 |
imantics
|
| 9 |
terminaltables
|
| 10 |
pycocotools
|
| 11 |
+
mmdet==2.16.0
|