【OCR Issue No.9】以可选形式支持Visualdl (#11947)
* delete visual dl * totally delete visual * delete vdl file * fix codestylepull/12025/head
parent
f7117efd44
commit
a730065e7b
|
@ -11,3 +11,4 @@ scikit-image
|
||||||
Shapely
|
Shapely
|
||||||
tqdm
|
tqdm
|
||||||
addict
|
addict
|
||||||
|
visualdl
|
||||||
|
|
|
@ -1,3 +1,2 @@
|
||||||
from .vdl_logger import VDLLogger
|
|
||||||
from .wandb_logger import WandbLogger
|
from .wandb_logger import WandbLogger
|
||||||
from .loggers import Loggers
|
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
|
lmdb
|
||||||
tqdm
|
tqdm
|
||||||
numpy
|
numpy
|
||||||
visualdl
|
|
||||||
rapidfuzz
|
rapidfuzz
|
||||||
opencv-python<=4.6.0.66
|
opencv-python<=4.6.0.66
|
||||||
opencv-contrib-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.save_load import save_model
|
||||||
from ppocr.utils.utility import print_dict, AverageMeter
|
from ppocr.utils.utility import print_dict, AverageMeter
|
||||||
from ppocr.utils.logging import get_logger
|
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.utils import profiler
|
||||||
from ppocr.data import build_dataloader
|
from ppocr.data import build_dataloader
|
||||||
|
|
||||||
|
@ -796,10 +796,11 @@ def preprocess(is_train=False):
|
||||||
loggers = []
|
loggers = []
|
||||||
|
|
||||||
if "use_visualdl" in config["Global"] and config["Global"]["use_visualdl"]:
|
if "use_visualdl" in config["Global"] and config["Global"]["use_visualdl"]:
|
||||||
save_model_dir = config["Global"]["save_model_dir"]
|
logger.warning(
|
||||||
vdl_writer_path = save_model_dir
|
"You are using VisualDL, the VisualDL is deprecated and "
|
||||||
log_writer = VDLLogger(vdl_writer_path)
|
"removed in ppocr!"
|
||||||
loggers.append(log_writer)
|
)
|
||||||
|
log_writer = None
|
||||||
if (
|
if (
|
||||||
"use_wandb" in config["Global"] and config["Global"]["use_wandb"]
|
"use_wandb" in config["Global"] and config["Global"]["use_wandb"]
|
||||||
) or "wandb" in config:
|
) or "wandb" in config:
|
||||||
|
|
Loading…
Reference in New Issue