├── .gitignore ├── LICENSE.md ├── README.md ├── base ├── __init__.py ├── base_dataset.py └── base_trainer.py ├── config ├── SynthText.yaml ├── SynthText_resnet18_FPN_DBhead_polyLR.yaml ├── bill2020.yaml ├── bill2020_MobileNetV3_FPN_DBhead_polyLR.yaml ├── bill2020_resnet18_FPN_DBhead_polyLR.yaml ├── icdar2015.yaml ├── icdar2015_dcn_resnet18_FPN_DBhead_polyLR.yaml ├── icdar2015_resnet18_FPN_DBhead_polyLR.yaml ├── icdar2015_resnet18_FPN_DBhead_polyLR_finetune.yaml ├── open_dataset.yaml ├── open_dataset_dcn_resnet50_FPN_DBhead_polyLR.yaml ├── open_dataset_resnest50_FPN_DBhead_polyLR.yaml └── open_dataset_resnet18_FPN_DBhead_polyLR.yaml ├── data_loader ├── __init__.py ├── dataset.py └── modules │ ├── __init__.py │ ├── augment.py │ ├── iaa_augment.py │ ├── make_border_map.py │ ├── make_shrink_map.py │ └── random_crop_data.py ├── datasets ├── test.txt ├── test │ ├── gt │ │ └── README.MD │ └── img │ │ └── README.MD ├── train.txt └── train │ ├── gt │ └── README.MD │ └── img │ └── README.MD ├── deploy ├── DBNet_C++_sample │ ├── CMakeLists.txt │ ├── db_detector.cpp │ ├── db_detector.h │ ├── main.cpp │ └── utils │ │ ├── CMakeLists.txt │ │ ├── Logger.cpp │ │ ├── Logger.h │ │ ├── clipper │ │ ├── clipper.cpp │ │ └── clipper.hpp │ │ ├── common.h │ │ ├── filesystem.cpp │ │ ├── filesystem.h │ │ ├── json.hpp │ │ ├── lanms.h │ │ ├── python_like_func.h │ │ ├── simpleini │ │ ├── CMakeLists.txt │ │ ├── ConvertUTF.c │ │ ├── ConvertUTF.h │ │ ├── LICENCE.txt │ │ ├── Makefile │ │ ├── README.md │ │ ├── ini.syn │ │ ├── simpleini.doxy │ │ ├── simpleini.h │ │ └── snippets.cpp │ │ ├── spdlog │ │ ├── async.h │ │ ├── async_logger.h │ │ ├── common.h │ │ ├── details │ │ │ ├── async_logger_impl.h │ │ │ ├── circular_q.h │ │ │ ├── console_globals.h │ │ │ ├── file_helper.h │ │ │ ├── fmt_helper.h │ │ │ ├── log_msg.h │ │ │ ├── logger_impl.h │ │ │ ├── mpmc_blocking_q.h │ │ │ ├── null_mutex.h │ │ │ ├── os.h │ │ │ ├── pattern_formatter.h │ │ │ ├── periodic_worker.h │ │ │ ├── registry.h │ │ │ └── thread_pool.h │ │ ├── fmt │ │ │ ├── bin_to_hex.h │ │ │ ├── bundled │ │ │ │ ├── LICENSE.rst │ │ │ │ ├── chrono.h │ │ │ │ ├── color.h │ │ │ │ ├── core.h │ │ │ │ ├── format-inl.h │ │ │ │ ├── format.h │ │ │ │ ├── locale.h │ │ │ │ ├── ostream.h │ │ │ │ ├── posix.h │ │ │ │ ├── printf.h │ │ │ │ ├── ranges.h │ │ │ │ └── time.h │ │ │ ├── fmt.h │ │ │ └── ostr.h │ │ ├── formatter.h │ │ ├── logger.h │ │ ├── sinks │ │ │ ├── android_sink.h │ │ │ ├── ansicolor_sink.h │ │ │ ├── base_sink.h │ │ │ ├── basic_file_sink.h │ │ │ ├── daily_file_sink.h │ │ │ ├── dist_sink.h │ │ │ ├── msvc_sink.h │ │ │ ├── null_sink.h │ │ │ ├── ostream_sink.h │ │ │ ├── rotating_file_sink.h │ │ │ ├── sink.h │ │ │ ├── stdout_color_sinks.h │ │ │ ├── stdout_sinks.h │ │ │ ├── syslog_sink.h │ │ │ └── wincolor_sink.h │ │ ├── spdlog.h │ │ ├── tweakme.h │ │ └── version.h │ │ ├── timing.cpp │ │ ├── timing.h │ │ ├── waybillNbCheck.h │ │ └── windows │ │ ├── dirent.h │ │ └── unistd_partial.h ├── common.py ├── infer_on_trt.py ├── infer_on_trt_dynamic.py └── readme.md ├── environment.yml ├── eval.sh ├── generate_lists.sh ├── imgs └── paper │ └── db.jpg ├── models ├── __init__.py ├── backbone │ ├── MobilenetV3.py │ ├── __init__.py │ ├── resnest │ │ ├── __init__.py │ │ ├── ablation.py │ │ ├── resnest.py │ │ ├── resnet.py │ │ └── splat.py │ ├── resnet.py │ └── shufflenetv2.py ├── basic.py ├── head │ ├── ConvHead.py │ ├── DBHead.py │ └── __init__.py ├── losses │ ├── DB_loss.py │ ├── __init__.py │ └── basic_loss.py ├── model.py └── neck │ ├── FPEM_FFM.py │ ├── FPN.py │ └── __init__.py ├── multi_gpu_train.sh ├── post_processing ├── __init__.py └── seg_detector_representer.py ├── predict.sh ├── requirement.txt ├── singlel_gpu_train.sh ├── test └── README.MD ├── tools ├── __init__.py ├── eval.py ├── predict.py └── train.py ├── trainer ├── __init__.py └── trainer.py └── utils ├── __init__.py ├── cal_recall ├── __init__.py ├── rrc_evaluation_funcs.py └── script.py ├── compute_mean_std.py ├── make_trainfile.py ├── metrics.py ├── ocr_metric ├── __init__.py └── icdar2015 │ ├── __init__.py │ ├── detection │ ├── __init__.py │ ├── deteval.py │ ├── icdar2013.py │ ├── iou.py │ └── mtwi2018.py │ └── quad_metric.py ├── schedulers.py └── util.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | .idea 3 | -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/README.md -------------------------------------------------------------------------------- /base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/base/__init__.py -------------------------------------------------------------------------------- /base/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/base/base_dataset.py -------------------------------------------------------------------------------- /base/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/base/base_trainer.py -------------------------------------------------------------------------------- /config/SynthText.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/SynthText.yaml -------------------------------------------------------------------------------- /config/SynthText_resnet18_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/SynthText_resnet18_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/bill2020.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/bill2020.yaml -------------------------------------------------------------------------------- /config/bill2020_MobileNetV3_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/bill2020_MobileNetV3_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/bill2020_resnet18_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/bill2020_resnet18_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/icdar2015.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/icdar2015.yaml -------------------------------------------------------------------------------- /config/icdar2015_dcn_resnet18_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/icdar2015_dcn_resnet18_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/icdar2015_resnet18_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/icdar2015_resnet18_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/icdar2015_resnet18_FPN_DBhead_polyLR_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/icdar2015_resnet18_FPN_DBhead_polyLR_finetune.yaml -------------------------------------------------------------------------------- /config/open_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/open_dataset.yaml -------------------------------------------------------------------------------- /config/open_dataset_dcn_resnet50_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/open_dataset_dcn_resnet50_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/open_dataset_resnest50_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/open_dataset_resnest50_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/open_dataset_resnet18_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/config/open_dataset_resnet18_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /data_loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/data_loader/__init__.py -------------------------------------------------------------------------------- /data_loader/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/data_loader/dataset.py -------------------------------------------------------------------------------- /data_loader/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/data_loader/modules/__init__.py -------------------------------------------------------------------------------- /data_loader/modules/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/data_loader/modules/augment.py -------------------------------------------------------------------------------- /data_loader/modules/iaa_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/data_loader/modules/iaa_augment.py -------------------------------------------------------------------------------- /data_loader/modules/make_border_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/data_loader/modules/make_border_map.py -------------------------------------------------------------------------------- /data_loader/modules/make_shrink_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/data_loader/modules/make_shrink_map.py -------------------------------------------------------------------------------- /data_loader/modules/random_crop_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/data_loader/modules/random_crop_data.py -------------------------------------------------------------------------------- /datasets/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/datasets/test.txt -------------------------------------------------------------------------------- /datasets/test/gt/README.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/datasets/test/gt/README.MD -------------------------------------------------------------------------------- /datasets/test/img/README.MD: -------------------------------------------------------------------------------- 1 | Place the `.jpg` files here. 2 | -------------------------------------------------------------------------------- /datasets/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/datasets/train.txt -------------------------------------------------------------------------------- /datasets/train/gt/README.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/datasets/train/gt/README.MD -------------------------------------------------------------------------------- /datasets/train/img/README.MD: -------------------------------------------------------------------------------- 1 | Place the `.jpg` files here. 2 | -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/CMakeLists.txt -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/db_detector.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/db_detector.cpp -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/db_detector.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/db_detector.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/main.cpp -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/CMakeLists.txt -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/Logger.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/Logger.cpp -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/Logger.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/Logger.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/clipper/clipper.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/clipper/clipper.cpp -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/clipper/clipper.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/clipper/clipper.hpp -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/common.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/filesystem.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/filesystem.cpp -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/filesystem.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/filesystem.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/json.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/json.hpp -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/lanms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/lanms.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/python_like_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/python_like_func.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/simpleini/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/simpleini/CMakeLists.txt -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/simpleini/ConvertUTF.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/simpleini/ConvertUTF.c -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/simpleini/ConvertUTF.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/simpleini/ConvertUTF.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/simpleini/LICENCE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/simpleini/LICENCE.txt -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/simpleini/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/simpleini/Makefile -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/simpleini/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/simpleini/README.md -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/simpleini/ini.syn: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/simpleini/ini.syn -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/simpleini/simpleini.doxy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/simpleini/simpleini.doxy -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/simpleini/simpleini.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/simpleini/simpleini.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/simpleini/snippets.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/simpleini/snippets.cpp -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/async.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/async.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/async_logger.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/async_logger.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/common.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/async_logger_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/async_logger_impl.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/circular_q.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/circular_q.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/console_globals.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/console_globals.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/file_helper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/file_helper.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/fmt_helper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/fmt_helper.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/log_msg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/log_msg.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/logger_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/logger_impl.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/mpmc_blocking_q.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/mpmc_blocking_q.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/null_mutex.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/null_mutex.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/os.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/os.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/pattern_formatter.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/pattern_formatter.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/periodic_worker.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/periodic_worker.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/registry.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/registry.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/details/thread_pool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/details/thread_pool.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bin_to_hex.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bin_to_hex.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/LICENSE.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/LICENSE.rst -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/chrono.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/chrono.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/color.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/color.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/core.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/core.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/format-inl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/format-inl.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/format.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/format.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/locale.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/locale.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/ostream.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/ostream.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/posix.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/posix.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/printf.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/printf.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/ranges.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/ranges.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/time.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/bundled/time.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/fmt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/fmt.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/fmt/ostr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/fmt/ostr.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/formatter.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/formatter.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/logger.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/logger.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/android_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/android_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/ansicolor_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/ansicolor_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/base_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/base_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/basic_file_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/basic_file_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/daily_file_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/daily_file_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/dist_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/dist_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/msvc_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/msvc_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/null_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/null_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/ostream_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/ostream_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/rotating_file_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/rotating_file_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/stdout_color_sinks.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/stdout_color_sinks.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/stdout_sinks.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/stdout_sinks.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/syslog_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/syslog_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/sinks/wincolor_sink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/sinks/wincolor_sink.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/spdlog.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/spdlog.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/tweakme.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/tweakme.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/spdlog/version.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/spdlog/version.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/timing.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/timing.cpp -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/timing.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/timing.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/waybillNbCheck.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/waybillNbCheck.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/windows/dirent.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/windows/dirent.h -------------------------------------------------------------------------------- /deploy/DBNet_C++_sample/utils/windows/unistd_partial.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/DBNet_C++_sample/utils/windows/unistd_partial.h -------------------------------------------------------------------------------- /deploy/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/common.py -------------------------------------------------------------------------------- /deploy/infer_on_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/infer_on_trt.py -------------------------------------------------------------------------------- /deploy/infer_on_trt_dynamic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/infer_on_trt_dynamic.py -------------------------------------------------------------------------------- /deploy/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/deploy/readme.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/environment.yml -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=0 python3 tools/eval.py --model_path '' -------------------------------------------------------------------------------- /generate_lists.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/generate_lists.sh -------------------------------------------------------------------------------- /imgs/paper/db.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/imgs/paper/db.jpg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbone/MobilenetV3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/backbone/MobilenetV3.py -------------------------------------------------------------------------------- /models/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/backbone/__init__.py -------------------------------------------------------------------------------- /models/backbone/resnest/__init__.py: -------------------------------------------------------------------------------- 1 | from .resnest import * -------------------------------------------------------------------------------- /models/backbone/resnest/ablation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/backbone/resnest/ablation.py -------------------------------------------------------------------------------- /models/backbone/resnest/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/backbone/resnest/resnest.py -------------------------------------------------------------------------------- /models/backbone/resnest/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/backbone/resnest/resnet.py -------------------------------------------------------------------------------- /models/backbone/resnest/splat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/backbone/resnest/splat.py -------------------------------------------------------------------------------- /models/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/backbone/resnet.py -------------------------------------------------------------------------------- /models/backbone/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/backbone/shufflenetv2.py -------------------------------------------------------------------------------- /models/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/basic.py -------------------------------------------------------------------------------- /models/head/ConvHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/head/ConvHead.py -------------------------------------------------------------------------------- /models/head/DBHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/head/DBHead.py -------------------------------------------------------------------------------- /models/head/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/head/__init__.py -------------------------------------------------------------------------------- /models/losses/DB_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/losses/DB_loss.py -------------------------------------------------------------------------------- /models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/losses/__init__.py -------------------------------------------------------------------------------- /models/losses/basic_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/losses/basic_loss.py -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/model.py -------------------------------------------------------------------------------- /models/neck/FPEM_FFM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/neck/FPEM_FFM.py -------------------------------------------------------------------------------- /models/neck/FPN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/neck/FPN.py -------------------------------------------------------------------------------- /models/neck/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/models/neck/__init__.py -------------------------------------------------------------------------------- /multi_gpu_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/multi_gpu_train.sh -------------------------------------------------------------------------------- /post_processing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/post_processing/__init__.py -------------------------------------------------------------------------------- /post_processing/seg_detector_representer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/post_processing/seg_detector_representer.py -------------------------------------------------------------------------------- /predict.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/predict.sh -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/requirement.txt -------------------------------------------------------------------------------- /singlel_gpu_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/singlel_gpu_train.sh -------------------------------------------------------------------------------- /test/README.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/test/README.MD -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/tools/predict.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/tools/train.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/trainer/__init__.py -------------------------------------------------------------------------------- /trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/trainer/trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/cal_recall/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/cal_recall/__init__.py -------------------------------------------------------------------------------- /utils/cal_recall/rrc_evaluation_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/cal_recall/rrc_evaluation_funcs.py -------------------------------------------------------------------------------- /utils/cal_recall/script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/cal_recall/script.py -------------------------------------------------------------------------------- /utils/compute_mean_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/compute_mean_std.py -------------------------------------------------------------------------------- /utils/make_trainfile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/make_trainfile.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/ocr_metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/ocr_metric/__init__.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/ocr_metric/icdar2015/__init__.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/detection/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/detection/deteval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/ocr_metric/icdar2015/detection/deteval.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/detection/icdar2013.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/ocr_metric/icdar2015/detection/icdar2013.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/detection/iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/ocr_metric/icdar2015/detection/iou.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/detection/mtwi2018.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/ocr_metric/icdar2015/detection/mtwi2018.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/quad_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/ocr_metric/icdar2015/quad_metric.py -------------------------------------------------------------------------------- /utils/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/schedulers.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangjinsong3/DBNet.tensorRT/HEAD/utils/util.py --------------------------------------------------------------------------------