Commit 244e87c4 authored by Yuxin Wu's avatar Yuxin Wu

fix evaluation

parent 1554550d
...@@ -114,7 +114,7 @@ def eval_coco(df, detect_func): ...@@ -114,7 +114,7 @@ def eval_coco(df, detect_func):
'image_id': img_id, 'image_id': img_id,
'category_id': cat_id, 'category_id': cat_id,
'bbox': list(map(lambda x: round(float(x), 2), box)), 'bbox': list(map(lambda x: round(float(x), 2), box)),
'score': round(r.score, 3), 'score': round(float(r.score), 3),
} }
# also append segmentation to results # also append segmentation to results
......
...@@ -122,9 +122,6 @@ class Monitors(Callback): ...@@ -122,9 +122,6 @@ class Monitors(Callback):
for m in self._monitors: for m in self._monitors:
assert isinstance(m, TrainingMonitor), m assert isinstance(m, TrainingMonitor), m
def _setup_graph(self):
self._scalar_history.setup_graph(self.trainer)
def _dispatch(self, func): def _dispatch(self, func):
for m in self._monitors: for m in self._monitors:
func(m) func(m)
...@@ -404,8 +401,6 @@ class ScalarPrinter(TrainingMonitor): ...@@ -404,8 +401,6 @@ class ScalarPrinter(TrainingMonitor):
self._enable_step = enable_step self._enable_step = enable_step
self._enable_epoch = enable_epoch self._enable_epoch = enable_epoch
def _setup_graph(self):
self._dic = {} self._dic = {}
def _trigger_step(self): def _trigger_step(self):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment