pytorch-yolo训练自己的数据集

Posted by kevin on March 20, 2020

preface

yolo 是一种目标检测算法,官方是基于 darknet 这种框架来训练的,darknet 是用 C 写的,有些硬核,所以我在 GitHub 上找到了人家用 pytorch 复现的 yolo,这次就拿 pytorch 结合 yolo 来训练一下自己的目标检测数据集

待续

训练好了,用了 22 个小时

train_over