Skip to content

Commit

Permalink
【OCR Issue No.9】以可选形式支持Visualdl (#11947)
Browse files Browse the repository at this point in the history
* delete visual dl

* totally delete visual

* delete vdl file

* fix codestyle
  • Loading branch information
Liyulingyue authored Apr 25, 2024
1 parent f7117ef commit a730065
Show file tree
Hide file tree
Showing 5 changed files with 8 additions and 30 deletions.
3 changes: 2 additions & 1 deletion benchmark/PaddleOCR_DBNet/requirement.txt
Original file line number Diff line number Diff line change
Expand Up @@ -10,4 +10,5 @@ PyYAML
scikit-image
Shapely
tqdm
addict
addict
visualdl
1 change: 0 additions & 1 deletion ppocr/utils/loggers/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,2 @@
from .vdl_logger import VDLLogger
from .wandb_logger import WandbLogger
from .loggers import Loggers
22 changes: 0 additions & 22 deletions ppocr/utils/loggers/vdl_logger.py

This file was deleted.

1 change: 0 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@ pyclipper
lmdb
tqdm
numpy
visualdl
rapidfuzz
opencv-python<=4.6.0.66
opencv-contrib-python<=4.6.0.66
Expand Down
11 changes: 6 additions & 5 deletions tools/program.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@
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

Expand Down Expand Up @@ -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:
Expand Down

0 comments on commit a730065

Please sign in to comment.