개정판 6e1bfbcb
issue #1366: testing
Change-Id: I58f667667d50ae81ed31a276879c530162c12f33
DTI_PID/WebServer/CRAFT_pytorch_master/text_craft.py | ||
---|---|---|
202 | 202 |
parser.add_argument('--text_threshold', default=0.7, type=float, help='text confidence threshold') |
203 | 203 |
parser.add_argument('--low_text', default=0.4, type=float, help='text low-bound score') |
204 | 204 |
parser.add_argument('--link_threshold', default=0.4, type=float, help='link confidence threshold') |
205 |
parser.add_argument('--cuda', default=False, type=str2bool, help='Use cuda for inference')
|
|
205 |
parser.add_argument('--cuda', default=True, type=str2bool, help='Use cuda for inference')
|
|
206 | 206 |
parser.add_argument('--canvas_size', default=1000, type=int, help='image size for inference') |
207 | 207 |
parser.add_argument('--mag_ratio', default=1, type=float, help='image magnification ratio') |
208 | 208 |
parser.add_argument('--poly', default=False, action='store_true', help='enable polygon type') |
DTI_PID/WebServer/app.py | ||
---|---|---|
18 | 18 |
import text_craft |
19 | 19 |
from service_streamer import ThreadedStreamer |
20 | 20 |
|
21 |
# for error at 3.8 |
|
22 |
import ctypes |
|
23 |
ctypes.cdll.LoadLibrary('caffe2_nvrtc.dll') |
|
24 |
|
|
21 | 25 |
streamer = ThreadedStreamer(text_craft.get_text_box_batch, batch_size=64) |
22 | 26 |
except ImportError as ex: |
23 | 27 |
ex |
DTI_PID/WebServer/service_streamer_master/service_streamer/service_streamer.py | ||
---|---|---|
17 | 17 |
|
18 | 18 |
TIMEOUT = 1 |
19 | 19 |
TIME_SLEEP = 0.001 |
20 |
WORKER_TIMEOUT = 2000
|
|
20 |
WORKER_TIMEOUT = 200 |
|
21 | 21 |
logger = logging.getLogger(__name__) |
22 | 22 |
logger.setLevel("INFO") |
23 | 23 |
|
내보내기 Unified diff