更新加入料带目标检测,判断料带到位,以及控制滚筒逻辑

This commit is contained in:
琉璃月光
2026-01-04 13:36:07 +08:00
parent ebcd7dcdcb
commit 68b14a4bec
69 changed files with 13 additions and 14 deletions

8
ailai_pc/test_model.py Normal file
View 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}")