未验证 提交 40c6f570 编写于 作者: B Bubbliiiing 提交者: GitHub

Update test.py

上级 4f2ddfe2
#--------------------------------------------#
# 该部分代码只用于看网络结构,并非测试代码
# map测试请看get_dr_txt.py、get_gt_txt.py
# 和get_map.py
#--------------------------------------------#
import torch
from nets.efficientnet import EfficientNet
from nets.efficientdet import EfficientDetBackbone
......@@ -19,4 +24,4 @@ if __name__ == '__main__':
# Test inference
model = EfficientDetBackbone(80,0)
print('# generator parameters:', sum(param.numel() for param in model.parameters()))
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册