利用netron查看网络框架

 加载模型，存成.jit格式，再把后缀改成.pt

from models import yolo
import torch
from PIL import Image,ImageDraw
from torchvision import transforms
from utils import utils
import torch.jit
from models.experimental import attempt_load

model = yolo.Model("models/yolov3.yaml")
model.load_state_dict(torch.load("runs/train/exp/weights/best.pt"))
model.eval()
img = Image.open(r"data/images/zidane.jpg")
tf = transforms.Compose([
    transforms.Resize((512,640)),
    transforms.ToTensor()
])
img_tensor = tf(img)
pred = model(img_tensor[None])[0]
script_model = torch.jit.trace(model,img_tensor[None])
script_model.save("my_yolov3.jit")

运行下面的代码，就可以在http://localhost:8080/这个网址上查看模型了

import netron
modelPath = "my_yolov3.pt"
netron.start(modelPath)