| 19 | |
| 20 | |
| 21 | def make_parser(): |
| 22 | parser = argparse.ArgumentParser("ByteTrack Demo!") |
| 23 | parser.add_argument( |
| 24 | "demo", default="image", help="demo type, eg. image, video and webcam" |
| 25 | ) |
| 26 | parser.add_argument("-expn", "--experiment-name", type=str, default=None) |
| 27 | parser.add_argument("-n", "--name", type=str, default=None, help="model name") |
| 28 | |
| 29 | parser.add_argument( |
| 30 | #"--path", default="./datasets/mot/train/MOT17-05-FRCNN/img1", help="path to images or video" |
| 31 | "--path", default="./videos/palace.mp4", help="path to images or video" |
| 32 | ) |
| 33 | parser.add_argument("--camid", type=int, default=0, help="webcam demo camera id") |
| 34 | parser.add_argument( |
| 35 | "--save_result", |
| 36 | action="store_true", |
| 37 | help="whether to save the inference result of image/video", |
| 38 | ) |
| 39 | |
| 40 | # exp file |
| 41 | parser.add_argument( |
| 42 | "-f", |
| 43 | "--exp_file", |
| 44 | default=None, |
| 45 | type=str, |
| 46 | help="pls input your expriment description file", |
| 47 | ) |
| 48 | parser.add_argument("-c", "--ckpt", default=None, type=str, help="ckpt for eval") |
| 49 | parser.add_argument( |
| 50 | "--device", |
| 51 | default="gpu", |
| 52 | type=str, |
| 53 | help="device to run our model, can either be cpu or gpu", |
| 54 | ) |
| 55 | parser.add_argument("--conf", default=None, type=float, help="test conf") |
| 56 | parser.add_argument("--nms", default=None, type=float, help="test nms threshold") |
| 57 | parser.add_argument("--tsize", default=None, type=int, help="test img size") |
| 58 | parser.add_argument("--fps", default=30, type=int, help="frame rate (fps)") |
| 59 | parser.add_argument( |
| 60 | "--fp16", |
| 61 | dest="fp16", |
| 62 | default=False, |
| 63 | action="store_true", |
| 64 | help="Adopting mix precision evaluating.", |
| 65 | ) |
| 66 | parser.add_argument( |
| 67 | "--fuse", |
| 68 | dest="fuse", |
| 69 | default=False, |
| 70 | action="store_true", |
| 71 | help="Fuse conv and bn for testing.", |
| 72 | ) |
| 73 | parser.add_argument( |
| 74 | "--trt", |
| 75 | dest="trt", |
| 76 | default=False, |
| 77 | action="store_true", |
| 78 | help="Using TensorRT model for testing.", |