Repository Details
Shared by
HelloGitHub Rating
0 ratings
完整的目标检测项目
Past 6 days Received 0 stars ✨
Use arrow keys to explore dates, Home and End for the first and last date, Escape to dismiss.| Date | Star | Daily change |
|---|---|---|
| 周日 | 406 | — |
| 周一 | 406 | 0 |
| 周二 | 406 | 0 |
| 周三 | 406 | 0 |
| 周四 | 406 | 0 |
| 周五 | 406 | 0 |
406
Stars
Yes
Chinese
Python
Language
No
Active
1
Contributors
2
Issues
No
Organization
None
Latest
186
Forks
Apache-2.0
License
完整的目标检测项目。结构简洁明了,中文注释。适宜新手入门、目标检测任务参考,甚至直接基于本项目实现目标检测任务。示例代码:
```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