text: import torch
  from torchsummary import summary
  from nets.yolo3 import YoloBody
  from utils.config import Config
  
  if __name__ == "__main__":
      # 需要使用device来指定网络在GPU还是CPU运行
      device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
      config = {"model_params": {"backbone_name": "darknet_53"},"yolo": {"anchors": [[1,2,3],[2,3,4],[3,4,5]],"classes": 80}}
      m = YoloBody(config).to(device)
      summary(m, input_size=(3, 416, 416))
  