0

使用Nvidia Retinanet在自定义数据集上训练视网膜网络。每当我尝试运行训练命令时,我都会得到以下输出

输入命令:
root@5aff9c57da58:/workspace# odtk train odtk/data/retinanet_rn50fpn.pth --backbone ResNet50FPN
--images odtk/data/train2017-sample/data --annotations odtk/data/train2017-sample/train2017-sample。 json
--val-images odtk/data/val2017-sample/data --val-annotations odtk/data/val2017-sample/val2017-sample.json

输出:
注意!安装 ujson 可以加快注解的加载速度。
从retinanet_rn50fpn.pth 加载模型...
模型:RetinaNet
骨干网:ResNet50FPN
类:80,锚点:9
准备数据集...
加载器:pytorch
调整大小:[640, 1024],最大值:1333
设备:1 GPU
批次:2,精度: 混合
BBOX 类型:轴对齐
90000 次迭代的训练模型...
root@5aff9c57da58:/workspace#

什么也没有发生,训练脚本停止运行。还尝试了不同的参数,如类、迭代器等。

4

0 回答 0