Spaces:
Runtime error
Runtime error
mnauf commited on
Commit ·
4c1b13e
1
Parent(s): 2001d90
added device
Browse files- sample_solution.py +1 -1
sample_solution.py
CHANGED
|
@@ -227,7 +227,7 @@ def parse_opt():
|
|
| 227 |
parser.add_argument('--conf-thres', type=float, default=0.7, help='confidence threshold')
|
| 228 |
parser.add_argument('--iou-thres', type=float, default=0.45, help='NMS IoU threshold')
|
| 229 |
parser.add_argument('--max-det', type=int, default=1000, help='maximum detections per image')
|
| 230 |
-
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
|
| 231 |
parser.add_argument('--view-img', action='store_true', help='show results')
|
| 232 |
parser.add_argument('--save-txt', action='store_true', help='save results to *.txt')
|
| 233 |
parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')
|
|
|
|
| 227 |
parser.add_argument('--conf-thres', type=float, default=0.7, help='confidence threshold')
|
| 228 |
parser.add_argument('--iou-thres', type=float, default=0.45, help='NMS IoU threshold')
|
| 229 |
parser.add_argument('--max-det', type=int, default=1000, help='maximum detections per image')
|
| 230 |
+
parser.add_argument('--device', default='cpu', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
|
| 231 |
parser.add_argument('--view-img', action='store_true', help='show results')
|
| 232 |
parser.add_argument('--save-txt', action='store_true', help='save results to *.txt')
|
| 233 |
parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')
|