| from ultralytics import YOLO | |
| # 加载状态字典 | |
| #model = torch.load("/home/lab/LJ/wampee/ultralytics/yolov8n.pt") | |
| model = YOLO('yolov8n.pt') # 加载预训练模型 | |
| state_dict = model.state_dict() | |
| lens=0 | |
| for name, param in state_dict.items(): | |
| print(f"Parameter: {name}, Shape: {param.shape}") | |