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')
|