Xiangzhe Lu
Xiangzhe Lu
I also hope that there will be a pytorch or tensorflow version of this code.
现在数据集的格式为一个train.txt文件，里边包含图片存储的绝对路径还有label，路径与label由空格隔开。label为数字格式 你可以自己生成 或者将每个类别放置在一个文件夹下，所有的类别文件放置在一个train文件夹下，然后使用data/preprocess.py生成
You could find pre-trained model in models/__init__.py. You also could get these things from torchvision.
最后一层直接取出的结果没有经过softmax，可以用一个softmax得到归一化的概率值
对于第一个问题，你需要到torchvision的模型定义中，提取你需要的模型层的输出，我这里采用的时efficientnet，为了省事没有去处理别的模型，在程序文件的注释中已经指明了这个问题 对于第二个问题，这是因为img放置在了gpu上，而模型忘记了，这个在程序中已经修改过来，谢谢指出
[email protected]留一个邮箱吧
不好意思，我没太注意过这个问题，回头试一试