super-gradients

收录理由

做视觉的团队常常要在分类、检测、分割、姿态估计等不同任务之间切换多个代码库，维护成本不低。SuperGradients把这几类工作统一到一个API下，配合现成的训练配置，你只要给出一条数据集路径就能从命令行跑起训练，不用手动调超参数。模型库里有带COCO预训练权重的YOLO-NAS系列，训练完可以直接导出ONNX，后续用TensorRT或OpenVINO部署也顺畅。它不像黑盒训练工具那样什么都要包办，也不是需要自己拼装的PyTorch流水线——该给的控制都给到了，但不会让你陷入写胶水代码的泥潭。对于想复现视觉训练结果、又不想从头组装整套流程的团队来说，是个很实际的起点。