提交 d750aa7a 编写于 作者: W wangguanzhong 提交者: GitHub

minor fix (#2961)

上级 597145c2
...@@ -13,9 +13,6 @@ ...@@ -13,9 +13,6 @@
# limitations under the License. # limitations under the License.
import numpy as np import numpy as np
import matplotlib
matplotlib.use('Agg')
from pycocotools.coco import COCO from pycocotools.coco import COCO
import logging import logging
......
...@@ -22,6 +22,8 @@ import sys ...@@ -22,6 +22,8 @@ import sys
import json import json
import cv2 import cv2
import numpy as np import numpy as np
import matplotlib
matplotlib.use('Agg')
from pycocotools.coco import COCO from pycocotools.coco import COCO
from pycocotools.cocoeval import COCOeval from pycocotools.cocoeval import COCOeval
import pycocotools.mask as mask_util import pycocotools.mask as mask_util
......
...@@ -61,8 +61,5 @@ class TrainingStats(object): ...@@ -61,8 +61,5 @@ class TrainingStats(object):
def log(self, extras=None): def log(self, extras=None):
d = self.get(extras) d = self.get(extras)
strs = ', '.join( strs = ', '.join(str(dict({x: y})).strip('{}') for x, y in d.items())
str(dict({
x.encode('utf-8'): y
})).strip('{}') for x, y in d.items())
return strs return strs
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册