YOLOV5代码解析（小白系列）.pdf

YOLOV5代码解析（⼩⽩系列⼀） yolov5 代码脚本解析 1. train.py 因为本⼈太菜了，所以此处就通过⼀⾏⾏debug，然后先记录整个运⾏逻辑，然后后⾯再⼀点点增补每个trick的原理。 def main(opt): setlogging(RANK) if RANK in [-1,0]: print(colorstr(train: )+, .join (f{k}={v} for k,v in vars(opt).items())) check_git_status() check_requirements(exclude=[thop]) 这⾥的 RANK是默认为-1 然后就开启记录logging的函数 在 general.py def set_logging(rank=-1,verbose=True): logging.basicConfig( , level=logging.INFO if (verbose and rank in [-1,0]) else logging.WARN) 其余两个函数是检测是否做好依赖以及当前是否有搭建github仓库，当然我是没有开启的。。。。 def main(opt): setlogging(RANK) if RANK in [-1,0]: print(colorstr(train: )+, .join (f{k}={v} for k,v in vars(opt).items())) check_git_status() check_requirements(exclude=[thop]) #Resume wandb_run=check_wandb_resume(opt) #None 所以这⾥的 就不上了 # if opt.data,opt.cfg,opt.hyp=check_file(opt.data),check_file(opt.cfg),check_file(opt.hyp) 这⾥的check_file函数是为了判断传⼊的数据路径是否合法general.py 通常如果数据路径不对，就会触发else:的分⽀ def check_file(file): file=str(file) if Path(file).is_file() or file ==: #exists return file elif file.startswith((http:/,https:/)): #download url=str(Path(file)).replace(:/,://) file = Path(urllib.parse.unquote(file)).name.split(?)[0] #%2F == / print(fDownloading {url} to {file}...) torch.hub.download_url_to_file(url,file) assert Path(file).exists() and Path(file).stat().st_size0,fFile download failed: {url} return file else: files=glob.glob(./**/+file,recursive=True) # find file assert len(files),fFile not found:{file} assert len(files)==1, fMultiple files match {file},specify exact path:{files}} return files[0] def main(opt): setlogging(RANK) if RANK in [-1,0]: print(colorstr(train: )+, .join (f{k}={v} for k,v in vars(opt).items())) check_git_status() check_requirements(exclude=[thop]) #Resume wandb_run=check_wandb_resume(opt) #None 所以这⾥的 就不上了 # if opt.data,opt.cfg,opt.hyp