官方pytorch版本Baseline在线推理代码有误,测试结果只有0.02几,搞了半天才发现`目录名称`和`预测的argmax`不是对应,可以通过`train_dataset.class_to_idx`找到映射关系,进行修改。
或者直接使用我修改过的customize_service.py文件
customize_service.zip
2.89 KB,下载次数:75
