# coding:utf-8
import torch
from torchsummary import summary
from nets.yolo3 import YoloBody

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": 10}}
    m = YoloBody(config).to(device)
    summary(m, input_size=(3, 416, 416))

b2888c1e5f5ad0b74072b4d67163dd3ee365c258
<mask token>

<mask token>
if __name__ == '__main__':
    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': 10}}
    m = YoloBody(config).to(device)
    summary(m, input_size=(3, 416, 416))

import torch
from torchsummary import summary
from nets.yolo3 import YoloBody
if __name__ == '__main__':
    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': 10}}
    m = YoloBody(config).to(device)
    summary(m, input_size=(3, 416, 416))

# coding:utf-8
import torch
from torchsummary import summary
from nets.yolo3 import YoloBody

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": 10}}
    m = YoloBody(config).to(device)
    summary(m, input_size=(3, 416, 416))
