TWK2022/ObjectDetection

基于pytorch实现的目标检测训练框架，各个部分模块化，方便修改模型。包含目标检测模型、训练、验证、测试、剪枝再训练、wandb可视化、onnx导出。

Python