Refactor argparser printing to print_args() (#4850)
* Refactor argparser printing to `print_args()` * Cleanup
This commit is contained in:
1 parent
84bfa89236
commit
3732f9ac8a
6 files changed
+23
-18
No files matched your search
@@ -41,7 +41,8 @@ from models.experimental import attempt_load
|
||||
from models.yolo import Detect
|
||||
from utils.activations import SiLU
|
||||
from utils.datasets import LoadImages
|
||||
from utils.general import colorstr, check_dataset, check_img_size, check_requirements, file_size, set_logging, url2file
|
||||
from utils.general import colorstr, check_dataset, check_img_size, check_requirements, file_size, print_args, \
|
||||
set_logging, url2file
|
||||
from utils.torch_utils import select_device
|
||||
|
||||
|
||||
@@ -322,12 +323,12 @@ def parse_opt():
|
||||
default=['torchscript', 'onnx'],
|
||||
help='available formats are (torchscript, onnx, coreml, saved_model, pb, tflite, tfjs)')
|
||||
opt = parser.parse_args()
|
||||
print_args(FILE.stem, opt)
|
||||
return opt
|
||||
|
||||
|
||||
def main(opt):
|
||||
set_logging()
|
||||
print(colorstr('export: ') + ', '.join(f'{k}={v}' for k, v in vars(opt).items()))
|
||||
run(**vars(opt))
|
||||
|
||||
|
||||
|
||||
Reference in new issue
Block a user