使用方法: dataset放到/workspace下面 文件说明: 三段式训练,用于解决目标太小的问题。 python run_train_full.py
简单,直接使用 run_train.py 单一阶段训练 复杂: run_hight_train 两阶段训练,因为第一阶段是1280训练,再二阶段添加了0.5的缩放,让模型适应640-1920的分辨率 run_train_full 三阶段训练,因为第一阶段是1280训练,再二阶段添加了0.5的缩放,第三阶段用于在640分辨率下微调。
至此,实际推理速度成为毫秒级。1280下,cpu推理基本在2秒。现在cpu下为100MS,gpu下为10ms至30ms