从预训练的 COCO 权重开始训练新模型
python3 balloon.py train --dataset=/path/to/balloon/dataset --weights=coco
继续训练你之前训练过的模型
python3 balloon.py train --dataset=/path/to/balloon/dataset --weights=last
从 ImageNet 权重开始训练新模型
python3 balloon.py train --dataset=/path/to/balloon/dataset --weights=imagenet
中的代码balloon.py设置为训练 3K 步(30 个 epoch,每个 100 步),并使用 2 的批量大小。更新计划以满足您的需求。