import os
import torch
detect_model = torch.load( "/home/ykn/algorithm_system/flask_web/weight/traffic/detect.pt")['model']
print(detect_model.children())