下拉刷新
项目详情
navbar_avatar分享
repo_avatar
HelloGitHub 评分
0 人评分
完整的目标检测项目
开源Apache-2.0
认领
收藏
分享
404
星数
中文
Python
主语言
活跃
1
贡献者
2
Issues
组织
最新版本
187
Forks
Apache-2.0
协议
更多
完整的目标检测项目。结构简洁明了,中文注释。适宜新手入门、目标检测任务参考,甚至直接基于本项目实现目标检测任务。示例代码: ```python # 实例化模型:模型的具体各种参数在 Config 文件中进行配置 net = SSD(cfg) # 将模型移动到 GPU 上,cfg.DEVICE.MAINDEVICE 定义了模型所使用的主 GPU # 模型的参数更新会在主 GPU 上进行 net.to(cfg.DEVICE.MAINDEVICE) # 初始化训练器:训练器参数已通过 cfg 进行配置;也可传入参数进行配置(但不建议) trainer = Trainer(cfg, max_iter=None, batch_size=None, train_devices=None, model_save_step=None, model_save_root=None, vis = None, vis_step=None) # 训练器开始:在数据集上训练模型 trainer(net, train_dataset) ```
收录于:
第 45 期

评论

评分:
暂无精选评论