Cleanup TorchVision related functions (#18790)

This commit is contained in:
Mohammed Yasin 2025-01-21 18:41:05 +08:00 committed by GitHub
parent 066c5443f5
commit 5306a8cc1f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 8 additions and 13 deletions

View file

@ -6,18 +6,11 @@
# Parameters
nc: 10 # number of classes
scales: # model compound scaling constants, i.e. 'model=yolo11n-cls.yaml' will call yolo11-cls.yaml with scale 'n'
# [depth, width, max_channels]
n: [0.33, 0.25, 1024]
s: [0.33, 0.50, 1024]
m: [0.67, 0.75, 1024]
l: [1.00, 1.00, 1024]
x: [1.00, 1.25, 1024]
# ResNet18 backbone
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [512, "resnet18", "DEFAULT", True, 2]] # truncate two layers from the end
- [-1, 1, TorchVision, [512, resnet18, DEFAULT, True, 2]] # truncate two layers from the end
# YOLO11n head
head: