Delete test.py

c530cfe1 · Bubbliiiing · GitHub · 812dddaf · 812dddaf
隐藏空白更改
内联并排

Showing with 0 addition and 10 deletion

test.py test.py +0 -10

未找到文件。
--- a/test.py
+++ b/test.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')
-    model = YoloBody(3,20).to(device)
-    summary(model, input_size=(3, 416, 416))