Update P2-P7 models/hub variants (#6230)

* Update p2-p7 `models/hub` variants

* Update common.py

* AutoAnchor camelcase corrections
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-01-06 11:08:09 -08:00
1 parent ad565e31d2
commit 33a67b4918
7 files changed
+50 -9

No files matched your search

+1 -1
View File
@@ -306,7 +306,7 @@ class DetectMultiBackend(nn.Module):
if pt: # PyTorch
model = attempt_load(weights if isinstance(weights, list) else w, map_location=device)
stride = int(model.stride.max()) # model stride
stride = max(int(model.stride.max()), 32) # model stride
names = model.module.names if hasattr(model, 'module') else model.names # get class names
self.model = model # explicitly assign for to(), cpu(), cuda(), half()
elif jit: # TorchScript