Spaces:
Build error
Build error
Update main.py
Browse files
main.py
CHANGED
@@ -46,7 +46,7 @@ parser.add_argument('--trained_model', default='weights/craft_mlt_25k.pth', type
|
|
46 |
parser.add_argument('--text_threshold', default=0.7, type=float, help='text confidence threshold')
|
47 |
parser.add_argument('--low_text', default=0.4, type=float, help='text low-bound score')
|
48 |
parser.add_argument('--link_threshold', default=0.4, type=float, help='link confidence threshold')
|
49 |
-
parser.add_argument('--cuda', default=
|
50 |
parser.add_argument('--canvas_size', default=1280, type=int, help='image size for inference')
|
51 |
parser.add_argument('--mag_ratio', default=1.5, type=float, help='image magnification ratio')
|
52 |
parser.add_argument('--poly', default=False, action='store_true', help='enable polygon type')
|
|
|
46 |
parser.add_argument('--text_threshold', default=0.7, type=float, help='text confidence threshold')
|
47 |
parser.add_argument('--low_text', default=0.4, type=float, help='text low-bound score')
|
48 |
parser.add_argument('--link_threshold', default=0.4, type=float, help='link confidence threshold')
|
49 |
+
parser.add_argument('--cuda', default=False, type=str2bool, help='Use cuda for inference')
|
50 |
parser.add_argument('--canvas_size', default=1280, type=int, help='image size for inference')
|
51 |
parser.add_argument('--mag_ratio', default=1.5, type=float, help='image magnification ratio')
|
52 |
parser.add_argument('--poly', default=False, action='store_true', help='enable polygon type')
|