From 872d73daf5b8bcb4fd26afb194796fa42257b473 Mon Sep 17 00:00:00 2001
From: wangjingyeye <1025993141@qq.com>
Date: Thu, 30 Jun 2022 08:59:05 +0000
Subject: [PATCH] add db++

---
 tools/program.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/program.py b/tools/program.py
index d86be5ba..e8e72fdd 100755
--- a/tools/program.py
+++ b/tools/program.py
@@ -574,8 +574,8 @@ def preprocess(is_train=False):
     assert alg in [
         'EAST', 'DB', 'SAST', 'Rosetta', 'CRNN', 'STARNet', 'RARE', 'SRN',
         'CLS', 'PGNet', 'Distillation', 'NRTR', 'TableAttn', 'SAR', 'PSE',
-        'SEED', 'SDMGR', 'LayoutXLM', 'LayoutLM', 'PREN', 'FCE', 'SVTR',
-        'ViTSTR', 'ABINet', 'DB++'
+        'SEED', 'SDMGR', 'LayoutXLM', 'LayoutLM', 'LayoutLMv2', 'PREN', 'FCE',
+        'SVTR', 'ViTSTR', 'ABINet', 'DB++'
     ]
 
     if use_xpu:
-- 
GitLab