Skip to content
Snippets Groups Projects
Commit 76e3aea3 authored by Stephane Chavin's avatar Stephane Chavin
Browse files

name option improve

parent 4d5ff325
No related branches found
No related tags found
No related merge requests found
......@@ -81,7 +81,7 @@ def run(
visualize=False, # visualize features
update=False, # update all models
project=ROOT / 'runs/detect', # save results to project/name
name='exp', # save results to project/name
name=None, # save results to project/name
exist_ok=False, # existing project/name ok, do not increment
line_thickness=3, # bounding box thickness (pixels)
hide_labels=False, # hide labels
......@@ -98,8 +98,10 @@ def run(
screenshot = source.lower().startswith('screen')
if is_url and is_file:
source = check_file(source) # download
if not args.name:
project_name = input('Please enter the name of your project : ')
else:
project_name = args.name
date_now = date.today().strftime("%Y%m%d")
if sound:
......@@ -268,7 +270,7 @@ def parse_opt():
parser.add_argument('--visualize', action='store_true', help='visualize features')
parser.add_argument('--update', action='store_true', help='update all models')
parser.add_argument('--project', default=ROOT / 'runs/detect', help='save results to project/name')
parser.add_argument('--name', default='exp', help='save results to project/name')
parser.add_argument('--name', default=None, help='save results to project/name')
parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
parser.add_argument('--line-thickness', default=3, type=int, help='bounding box thickness (pixels)')
parser.add_argument('--hide-labels', default=False, action='store_true', help='hide labels')
......
......@@ -459,7 +459,7 @@ def parse_opt(known=False):
parser.add_argument('--sync-bn', action='store_true', help='use SyncBatchNorm, only available in DDP mode')
parser.add_argument('--workers', type=int, default=8, help='max dataloader workers (per RANK in DDP mode)')
parser.add_argument('--project', default=ROOT / 'runs/train', help='save to project/name')
parser.add_argument('--name', default='exp', help='save to project/name')
parser.add_argument('--name', default=None, help='save to project/name')
parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
parser.add_argument('--quad', action='store_true', help='quad dataloader')
parser.add_argument('--cos-lr', action='store_true', help='cosine LR scheduler')
......@@ -512,7 +512,10 @@ def main(opt, callbacks=Callbacks()):
opt.exist_ok, opt.resume = opt.resume, False # pass resume to exist_ok and disable resume
if opt.name == 'cfg':
opt.name = Path(opt.cfg).stem # use model.yaml as name
if not args.name:
project_name = input('Please enter the name of your project : ')
else:
project_name = args.name
date_now = date.today().strftime("%Y%m%d")
folder_name = '_'.join([project_name, date_now, opt.weights[-4:-3], str(opt.imgsz), opt.optimizer, opt.rf, opt.duration, 'YOLOV5',])
print(f'Your train results will be saved in {folder_name}')
......
......@@ -359,7 +359,7 @@ def parse_opt():
parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')
parser.add_argument('--save-json', action='store_true', help='save a COCO-JSON results file')
parser.add_argument('--project', default=ROOT / 'runs/val', help='save to project/name')
parser.add_argument('--name', default='exp', help='save to project/name')
parser.add_argument('--name', default=None, help='save to project/name')
parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
parser.add_argument('--half', action='store_true', help='use FP16 half-precision inference')
parser.add_argument('--dnn', action='store_true', help='use OpenCV DNN for ONNX inference')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment