diff --git a/test.py b/test.py deleted file mode 100644 index 32ef763b1c1749bcc4159a2901a7cc9615870158..0000000000000000000000000000000000000000 --- a/test.py +++ /dev/null @@ -1,13 +0,0 @@ -#--------------------------------------------# -# 该部分代码只用于看网络结构,并非测试代码 -# 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()))