pytorch/examples

How to use trained model to classifier pictures?

Open

#450 创建于 2018年11月25日

在 GitHub 查看
 (2 评论) (0 反应) (0 负责人)Python (9,429 fork)batch import
help wantedvision

仓库指标

Star
 (21,634 star)
PR 合并指标
 (30 天内没有已合并 PR)

描述

I have trained a best model by imagenet,but code repo has given does not have test option,so how can I use the model have trained to classifier pictures with labels?

贡献者指南

研究方向
编写一个小脚本，加载训练好的模型，预处理输入图像（例如调整大小、使用ImageNet统计信息进行归一化），运行推理，并将输出索引映射到ImageNet类别标签。使用PyTorch的torchvision.transforms进行预处理。
技术栈
pythonpytorch
领域
machine learningai
议题类型
调研
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
2
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
1-3 小时
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
活跃
清晰度议题是否清楚说明期望改动、验收标准和下一步。
清晰
前置要求
PythonPyTorch
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
70