下拉刷新
Repository Details
Shared bynavbar_avatar
repo_avatar
HelloGitHub Rating
0 ratings
完整的目标检测项目
FreeApache-2.0
Claim
Collect
Share
404
Stars
Yes
Chinese
Python
Language
No
Active
1
Contributors
2
Issues
No
Organization
None
Latest
187
Forks
Apache-2.0
License
More
完整的目标检测项目。结构简洁明了,中文注释。适宜新手入门、目标检测任务参考,甚至直接基于本项目实现目标检测任务。示例代码: ```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) ```
Included in:
Vol.45

Comments

Rating:
No comments yet