centerfusion专题

【CenterFusion】run_epoch()函数-训练一轮epoch-CenterFusion/src/lib/trainer.py

文件位置:CenterFusion/src/lib/trainer.py run_epoch作用:CenterFusion 项目训练一轮epoch过程 在 main.py 函数中,生成了训练器,然后再使用训练器训练一个 epochrun_epoch()函数的定义在src\lib\trainer.py150行左右,它的主要过程如下所示: def run_epoch(self, phase,

【CenterFusion】参数处理CenterFusion/src/lib/opts.py

文件内容:CenterFusion/src/lib/opts.py 文件作用:train.sh 脚本中参数的处理 首先对添加参数部分进行说明 例如: self.parser.add_argument('--not_set_cuda_env', action='store_true',help='used when training in slurm clusters.') action:脚

【CenterFusion】模型的创建、导入、保存CenterFusion/src/lib/model/model.py

文件内容:CenterFusion/src/lib/model/model.py 文件作用:模型的创建、导入、保存 model.py 具体内容如下: from __future__ import absolute_importfrom __future__ import divisionfrom __future__ import print_functionimport torchvis