Glenn Jocher
4 years ago
1 changed files with 1 additions and 3 deletions
+ 1
- 3
models/yolo.py
View File
| import argparse | import argparse | ||||
| import yaml | |||||
| from models.experimental import * | from models.experimental import * | ||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| parser = argparse.ArgumentParser() | parser = argparse.ArgumentParser() | ||||
| parser.add_argument('--cfg', type=str, default='yolov5s.yaml', help='model.yaml') | |||||
| parser.add_argument('--cfg', type=str, default='yolov5m-p6.yaml', help='model.yaml') | |||||
| parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu') | parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu') | ||||
| opt = parser.parse_args() | opt = parser.parse_args() | ||||
| opt.cfg = check_file(opt.cfg) # check file | opt.cfg = check_file(opt.cfg) # check file |
Loading…