【OCR Issue No.9】以可选形式支持Visualdl ()

* delete visual dl

* totally delete visual

* delete vdl file

* fix codestyle
pull/12025/head
张春乔 2024-04-25 17:37:27 +08:00 committed by GitHub
parent f7117efd44
commit a730065e7b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 8 additions and 30 deletions
benchmark/PaddleOCR_DBNet
ppocr/utils/loggers

View File

@ -10,4 +10,5 @@ PyYAML
scikit-image
Shapely
tqdm
addict
addict
visualdl

View File

@ -1,3 +1,2 @@
from .vdl_logger import VDLLogger
from .wandb_logger import WandbLogger
from .loggers import Loggers

View File

@ -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()

View File

@ -5,7 +5,6 @@ pyclipper
lmdb
tqdm
numpy
visualdl
rapidfuzz
opencv-python<=4.6.0.66
opencv-contrib-python<=4.6.0.66

View File

@ -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: