mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

trained

pull/13567/head
StrickenSun 2025-04-19 16:32:02 +08:00
parent 984b97ca01
commit ef6a71d51c
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
detect.py
View File

@ -266,13 +266,13 @@ def run(weights=ROOT / 'yolov5s.pt', # model.pt path(s)
def parse_opt():
parser = argparse.ArgumentParser()
parser.add_argument('--weights', nargs='+', type=str, default=ROOT / 'yolov5s.pt', help='model path(s)')
parser.add_argument('--source', type=str, default=ROOT / 'data/images', help='file/dir/URL/glob, 0 for webcam')
parser.add_argument('--weights', nargs='+', type=str, default=r'D:\lzy\yolov5\runs\train\exp6\weights\best.pt', help='model path(s)')
parser.add_argument('--source', type=str, default=r'D:\lzy\yolov5\data\yolo_dataset\test\images', help='file/dir/URL/glob, 0 for webcam')
parser.add_argument('--imgsz', '--img', '--img-size', nargs='+', type=int, default=[640], help='inference size h,w')
parser.add_argument('--conf-thres', type=float, default=0.25, help='confidence threshold')
parser.add_argument('--iou-thres', type=float, default=0.45, help='NMS IoU threshold')
parser.add_argument('--max-det', type=int, default=1000, help='maximum detections per image')
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')
parser.add_argument('--view-img', action='store_true', help='show results')
parser.add_argument('--save-txt', action='store_true', help='save results to *.txt')
parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')

6
train.py
View File

@ -433,11 +433,13 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
def parse_opt(known=False):
parser = argparse.ArgumentParser()
parser.add_argument('--weights', type=str, default=r'D:\lzy\yolov5\weight\yolov5n.pt', help='initial weights path')
parser.add_argument('--weights', type=str, default=r'D:\lzy\yolov5\runs\train\exp3\weights\best.pt', help='initial weights path')
parser.add_argument('--cfg', type=str, default=r'D:\lzy\yolov5\models\yolov5n.yaml', help='model.yaml path')
parser.add_argument('--data', type=str, default=r'D:\lzy\yolov5\data\my_data.yaml', help='dataset.yaml path')
parser.add_argument('--hyp', type=str, default=ROOT / 'data/hyps/hyp.scratch.yaml', help='hyperparameters path')
parser.add_argument('--epochs', type=int, default=300)
parser.add_argument('--epochs', type=int, default=150
)
parser.add_argument('--batch-size', type=int, default=1, help='total batch size for all GPUs')
parser.add_argument('--imgsz', '--img', '--img-size', type=int, default=640, help='train, val image size (pixels)')
parser.add_argument('--rect', action='store_true', help='rectangular training')