Ahsen Khaliq
commited on
Commit
·
124dee4
1
Parent(s):
40cdc23
Update demos/general.py
Browse files- demos/general.py +1 -1
demos/general.py
CHANGED
@@ -28,7 +28,7 @@ FPS_TEXT_SIZE = 2
|
|
28 |
if __name__ == '__main__':
|
29 |
parser = argparse.ArgumentParser()
|
30 |
parser.add_argument('--data', type=str, default='data/coco-kp.yaml')
|
31 |
-
parser.add_argument('--imgsz', type=int, default=
|
32 |
parser.add_argument('--vid', type=str, default='')
|
33 |
parser.add_argument('--weights', default='kapao_s_coco.pt')
|
34 |
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or cpu')
|
|
|
28 |
if __name__ == '__main__':
|
29 |
parser = argparse.ArgumentParser()
|
30 |
parser.add_argument('--data', type=str, default='data/coco-kp.yaml')
|
31 |
+
parser.add_argument('--imgsz', type=int, default=448)
|
32 |
parser.add_argument('--vid', type=str, default='')
|
33 |
parser.add_argument('--weights', default='kapao_s_coco.pt')
|
34 |
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or cpu')
|