Unverified 提交 dc13820c authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

TensorBoard `on_train_end()` speed improvements (#4605)

上级 b894e69d
...@@ -131,10 +131,9 @@ class Loggers(): ...@@ -131,10 +131,9 @@ class Loggers():
files = [(self.save_dir / f) for f in files if (self.save_dir / f).exists()] # filter files = [(self.save_dir / f) for f in files if (self.save_dir / f).exists()] # filter
if self.tb: if self.tb:
from PIL import Image import cv2
import numpy as np
for f in files: for f in files:
self.tb.add_image(f.stem, np.asarray(Image.open(f)), epoch, dataformats='HWC') self.tb.add_image(f.stem, cv2.imread(str(f))[..., ::-1], epoch, dataformats='HWC')
if self.wandb: if self.wandb:
self.wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]}) self.wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]})
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论