未验证 提交 39c776b9 编写于 作者: B Bubbliiiing 提交者: GitHub

Delete test.py

上级 83725f1c
#--------------------------------------------#
# 该部分代码只用于看网络结构,并非测试代码
# map测试请看get_dr_txt.py、get_gt_txt.py
# 和get_map.py
#--------------------------------------------#
import torch
from torchsummary import summary
from nets import ssd
if __name__ == "__main__":
model = ssd.get_ssd("train", 21)
print('# generator parameters:', sum(param.numel() for param in model.parameters()))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册