rknn替换,板子是3568的

This commit is contained in:
琉璃月光
2025-11-03 16:10:50 +08:00
parent d3a5cbfad0
commit 5b29081c06
23 changed files with 799 additions and 44 deletions

8
ailai_pc/lingshi.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}")