【OCR Issue No.9】以可选形式支持Visualdl (#11947)
* delete visual dl * totally delete visual * delete vdl file * fix codestylepull/12025/head
parent
f7117efd44
commit
a730065e7b
|
@ -10,4 +10,5 @@ PyYAML
|
|||
scikit-image
|
||||
Shapely
|
||||
tqdm
|
||||
addict
|
||||
addict
|
||||
visualdl
|
||||
|
|
|
@ -1,3 +1,2 @@
|
|||
from .vdl_logger import VDLLogger
|
||||
from .wandb_logger import WandbLogger
|
||||
from .loggers import Loggers
|
||||
|
|
|
@ -1,22 +0,0 @@
|
|||
from .base_logger import BaseLogger
|
||||
from visualdl import LogWriter
|
||||
|
||||
|
||||
class VDLLogger(BaseLogger):
|
||||
def __init__(self, save_dir):
|
||||
super().__init__(save_dir)
|
||||
self.vdl_writer = LogWriter(logdir=save_dir)
|
||||
|
||||
def log_metrics(self, metrics, prefix=None, step=None):
|
||||
if not prefix:
|
||||
prefix = ""
|
||||
updated_metrics = {prefix + "/" + k: v for k, v in metrics.items()}
|
||||
|
||||
for k, v in updated_metrics.items():
|
||||
self.vdl_writer.add_scalar(k, v, step)
|
||||
|
||||
def log_model(self, is_best, prefix, metadata=None):
|
||||
pass
|
||||
|
||||
def close(self):
|
||||
self.vdl_writer.close()
|
|
@ -5,7 +5,6 @@ pyclipper
|
|||
lmdb
|
||||
tqdm
|
||||
numpy
|
||||
visualdl
|
||||
rapidfuzz
|
||||
opencv-python<=4.6.0.66
|
||||
opencv-contrib-python<=4.6.0.66
|
||||
|
|
|
@ -33,7 +33,7 @@ from ppocr.utils.stats import TrainingStats
|
|||
from ppocr.utils.save_load import save_model
|
||||
from ppocr.utils.utility import print_dict, AverageMeter
|
||||
from ppocr.utils.logging import get_logger
|
||||
from ppocr.utils.loggers import VDLLogger, WandbLogger, Loggers
|
||||
from ppocr.utils.loggers import WandbLogger, Loggers
|
||||
from ppocr.utils import profiler
|
||||
from ppocr.data import build_dataloader
|
||||
|
||||
|
@ -796,10 +796,11 @@ def preprocess(is_train=False):
|
|||
loggers = []
|
||||
|
||||
if "use_visualdl" in config["Global"] and config["Global"]["use_visualdl"]:
|
||||
save_model_dir = config["Global"]["save_model_dir"]
|
||||
vdl_writer_path = save_model_dir
|
||||
log_writer = VDLLogger(vdl_writer_path)
|
||||
loggers.append(log_writer)
|
||||
logger.warning(
|
||||
"You are using VisualDL, the VisualDL is deprecated and "
|
||||
"removed in ppocr!"
|
||||
)
|
||||
log_writer = None
|
||||
if (
|
||||
"use_wandb" in config["Global"] and config["Global"]["use_wandb"]
|
||||
) or "wandb" in config:
|
||||
|
|
Loading…
Reference in New Issue