diff --git a/test.py b/test.py
index d2c53dac351451380a4228505b792da003403e1f..a669bbc40b24df98088fe711875a28b8f9dd008b 100644
--- a/test.py
+++ b/test.py
@@ -1,12 +1,13 @@
-import torch
-from torchsummary import summary
-from nets.CSPdarknet import darknet53
-from nets.yolo4 import YoloBody
 #--------------------------------------------#
 #   该部分代码只用于看网络结构，并非测试代码
 #   map测试请看get_dr_txt.py、get_gt_txt.py
 #   和get_map.py
 #--------------------------------------------#
+import torch
+from torchsummary import summary
+from nets.CSPdarknet import darknet53
+from nets.yolo4 import YoloBody
+
 if __name__ == "__main__":
     # 需要使用device来指定网络在GPU还是CPU运行
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')