Highwe2hell / resnet-50

该项目基于 ResNet-50 模型进行图像分类,使用 PyTorch 实现,支持图像预处理、数据增强、训练与验证过程,并提供提前停止机制以避免过拟合。用户可以使用该代码进行任意图像分类任务的训练和推理。
10Updated 4 months ago

Alternatives and similar repositories for resnet-50:

Users that are interested in resnet-50 are comparing it to the libraries listed below