"...resnet50_tensorflow.git" did not exist on "ad3ff44b1d6584ebc4d31ec92a21f9ee46629cc3"
Commit 22caa57c authored by Rayyyyy's avatar Rayyyyy
Browse files

Fix bug in test.py

parent 042ea171
...@@ -102,7 +102,7 @@ bash train.sh ...@@ -102,7 +102,7 @@ bash train.sh
## 推理 ## 推理
验证前需提前准备好预训练模型, 可从 参考资料 中提供的模型下载, 并将coco_path设置为当前环境中推理数据的对应地址, 数据应为COCO数据格式。 验证前需提前准备好预训练模型, 可从最下面的**参考资料**项目中提供的模型下载, 并将coco_path设置为当前环境中推理数据的对应地址, 数据应为COCO数据格式。
如果想要查看预测效果(预测结果输出到图片上), 请执行(其余参数如backbone等需与训练模型参数需一致, 详情请根据训练参数配置): 如果想要查看预测效果(预测结果输出到图片上), 请执行(其余参数如backbone等需与训练模型参数需一致, 详情请根据训练参数配置):
......
...@@ -116,7 +116,7 @@ if __name__ == "__main__": ...@@ -116,7 +116,7 @@ if __name__ == "__main__":
device = torch.device(args.device) device = torch.device(args.device)
if not os.path.exists(args.save_path): if not os.path.exists(args.save_path):
os.makedirs() os.makedirs(args.save_path)
model, postprocessors = build_test_model(args) model, postprocessors = build_test_model(args)
model.to(device) model.to(device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment