更新加入料带目标检测,判断料带到位,以及控制滚筒逻辑
This commit is contained in:
8
ailai_pc/test_model.py
Normal file
8
ailai_pc/test_model.py
Normal file
@ -0,0 +1,8 @@
|
||||
from ultralytics import YOLO
|
||||
import torch
|
||||
model = YOLO("/home/hx/开发/ailai_image_obb/ailai_pc/detect.pt")
|
||||
|
||||
for name, module in model.model.named_modules():
|
||||
if isinstance(module, torch.nn.Conv2d):
|
||||
w = module.weight
|
||||
print(f"{name} -> min: {w.min().item():.3f}, max: {w.max().item():.3f}")
|
||||
Reference in New Issue
Block a user