PyTorch
乱觉先森
这个作者很懒,什么都没留下…
展开
-
EfficientNet测试单张图片
EfficientNet 测试单张图片如何训练自己的数据集我在之前的博文:EfficientNet 训练测试自己的分类数据集介绍过。我也写过怎样在测试集算准确率。最近有小伙伴一直想知道怎么利用训练好的模型对一张图片进行预测类别标签。其实我前两个博文的代码真的理解了是很好自己修改的。思路就是把图片转化成Tensor然后送到训练好的模型forward一下。我觉得造成困扰的原因应该是不知道怎么找到自己数据集文件夹名字和预测的label之间的对应关系。1. 数据集文件夹的名字和训练好的Label_ID对应关系原创 2020-06-09 22:45:59 · 1695 阅读 · 11 评论 -
EfficientNet利用训练好的模型测试
EfficientNet 利用训练好的模型测试Google团队19年的EfficientNet和EfficientDet在图像分类和目标检测方面都取得了惊艳的效果。1. 训练过程EfficientNet如何训练自己的数据集我在之前的博文:EfficientNet 训练测试自己的分类数据集介绍过。不再赘述,通过训练我们已经可以在数据集下的model文件夹里保留最好的模型。2. 测试过程其实上次的代码稍微修改就可以利用保存的文件进行新的测试,你只需要修改main函数最前面的几行代码,下面放上完整代码:原创 2020-05-12 21:42:54 · 3086 阅读 · 10 评论 -
Pytorch把Tensor转化成图像可视化
Pytorch把Tensor转化成图像可视化在调试程序的时候经常想把tensor可视化成来看看,可以这样操作:from torchvision import transformsunloader = transforms.ToPILImage()image = original_tensor.cpu().clone() # clone the tensorimage = image.s...原创 2020-04-27 09:11:15 · 18745 阅读 · 3 评论 -
EfficientNet 训练测试自己的分类数据集
EfficientNet 训练测试自己的分类数据集Google团队19年的EfficientNet和EfficientDet在图像分类和目标检测方面都取得了惊艳的效果。EfficientNet既快又准,目测这两年将会成为比ResNet更广泛使用的Backbone。下面简单介绍在自己的分类数据集上运行EfficientNet。1. 准备代码Pytorch版本官方代码在这:EfficientNe...原创 2020-04-24 16:31:34 · 15002 阅读 · 111 评论