esun-choi commited on
Commit
65f195b
·
1 Parent(s): 160269e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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=True, 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')
 
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')