一个小的示例包
monk-test1的Python项目详细描述
项目详细信息
基于GluonCV Fintuning项目的管道-https://gluon-cv.mxnet.io/build/examples_detection/index.html
安装
支持
- Python 3.6
- Python 3.7
cd installation
使用命令检查cuda版本
nvcc -V
选择正确的需求文件并运行
cat <selected requirements file> | xargs -n 1 -L 1 pip install
例如cuda 9.0
cat requirements_cuda9.0.txt | xargs -n 1 -L 1 pip install
功能文件
管道
- 加载数据集
gtf.Dataset(root_dir, img_dir, anno_file, batch_size=batch_size);
- 负荷模型
gtf.Model(model_name, use_pretrained=pretrained, use_gpu=gpu);
- 设置超参数
gtf.Set_Learning_Rate(0.001);
- 火车
gtf.Train(epochs, params_file);
托多
- [x] 添加SSD支持
- [x] 添加YoloV3支持
- []添加对Coco类型注释数据集的支持
- [x] 添加对VOC类型注释数据集的支持
- []添加更快的RCNN支持
- [x] 对Kaggle和Colab的测试
- []添加验证功能和数据管道
- []添加优化器选择功能
- []启用学习速率计划程序支持
- []启用层冻结
- []设置详细级别
- []添加项目管理和版本控制支持(类似于Monk分类)
- []添加图形可视化支持
- []在推理时启用批处理处理
- []添加top-k输出可视化功能
- [x] 添加多GPU培训
- []自动更正丢失或损坏的图像-当前跳过这些图像
- []添加实验数据分析功能
外部贡献者列表
- 项目
标签: