lichen/ieemoo-ai-zhanting
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

3588适配

Browse Source
This commit is contained in:
李晨
2024-10-17 19:14:27 +08:00
parent 8475980895
commit a95db2a8fb
9 changed files with 78 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
train_zhanting.py Executable file → Normal file
View File

@ -1,7 +1,13 @@
import argparse
import logging
import math
import os
import os,sys
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
#print(os.path.dirname(os.path.abspath(__file__)))
#wandb
import wandb
#wandb.init(project="ieemoo-ai-zhanting", entity="wb_ht")
import random
import time
from copy import deepcopy
@ -35,10 +41,11 @@ from utils.plots import plot_images, plot_labels, plot_results, plot_evolution
from utils.torch_utils import ModelEMA, select_device, intersect_dicts, torch_distributed_zero_first, is_parallel
from utils.wandb_logging.wandb_utils import WandbLogger, check_wandb_resume
logger = logging.getLogger(__name__)
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
os.environ["CUDA_VISIBLE_DEVICES"] ="1"
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
def train(hyp, opt, device, tb_writer=None):
#logger.info(colorstr('hyperparameters: ') + ', '.join(f'{k}={v}'for k, v in hyp.items()))
save_dir, epochs, batch_size, total_batch_size, weights, rank = \
@ -459,16 +466,16 @@ def train(hyp, opt, device, tb_writer=None):
if __name__ == '__main__':
parser = argparse.ArgumentParser()
#parser.add_argument('--weights', type=str, default='runs/zhanting/yolov5s_finetune/exp9/weights/best.pt', help='initial weights path')
#parser.add_argument('--weights', type=str, default='runs/zhanting/yolov5m_finetune/exp/weights/best.pt', help='initial weights path')
parser.add_argument('--weights', type=str, default='weights/yolov5m.pt', help='initial weights path')
#parser.add_argument('--weights', type=str, default='runs/zhanting/yolov5m_finetune/exp10/weights/best.pt', help='initial weights path')
parser.add_argument('--weights', type=str, default='weights/yolov5l.pt', help='initial weights path')
#parser.add_argument('--weights', type=str, default='runs/zhanting/yolov5l_finetune/exp7/weights/best.pt', help='initial weights path')
parser.add_argument('--cfg', type=str, default='models/yolov5m.yaml', help='model.yaml path')
parser.add_argument('--cfg', type=str, default='models/yolov5l.yaml', help='model.yaml path')
parser.add_argument('--data', type=str, default='data/zhanting.yaml', help='data.yaml path')
parser.add_argument('--hyp', type=str, default='data/hyp.finetune.yaml', help='hyperparameters path')
#parser.add_argument('--hyp', type=str, default='data/hyp.scratch.yaml', help='hyperparameters path')
parser.add_argument('--epochs', type=int, default=600)
parser.add_argument('--batch-size', type=int, default=32, help='total batch size for all GPUs')
parser.add_argument('--img-size', nargs='+', type=int, default=[640, 640], help='[train, test] image sizes')
parser.add_argument('--epochs', type=int, default=200)
parser.add_argument('--batch-size', type=int, default=72, help='total batch size for all GPUs')
parser.add_argument('--img-size', nargs='+', type=int, default=[640,640], help='[train, test] image sizes')
parser.add_argument('--rect', action='store_true', help='rectangular training')
parser.add_argument('--resume', nargs='?', const=True, default=False, help='resume most recent training')
parser.add_argument('--nosave', action='store_true', help='only save final checkpoint')
@ -485,10 +492,10 @@ if __name__ == '__main__':
parser.add_argument('--sync-bn', action='store_true', help='use SyncBatchNorm, only available in DDP mode')
parser.add_argument('--local_rank', type=int, default=-1, help='DDP parameter, do not modify')
parser.add_argument('--workers', type=int, default=8, help='maximum number of dataloader workers')
#parser.add_argument('--project', default='runs/zhanting/yolov5s_finetune', help='save to project/name')
parser.add_argument('--project', default='runs/zhanting/yolov5m_finetune', help='save to project/name')
parser.add_argument('--project', default='runs/zhanting/yolov5l_finetune', help='save to project/name')
#parser.add_argument('--project', default='runs/zhanting/yolov5m_finetune', help='save to project/name')
#parser.add_argument('--project', default='runs/zhanting/yolov5_scratch', help='save to project/name')
parser.add_argument('--entity', default=None, help='W&B entity')
parser.add_argument('--entity', default="wb_ht", help='W&B entity')
parser.add_argument('--name', default='exp', 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')