narugo's picture
Publish model head_detect_v0.1_s, on 2024-09-16 04:21:57 UTC
b45eeae verified
raw
history blame
943 Bytes
{"names": ["head"], "yaml": {"nc": 1, "depth_multiple": 0.33, "width_multiple": 0.5, "backbone": [[-1, 1, "Conv", [64, 3, 2]], [-1, 1, "Conv", [128, 3, 2]], [-1, 3, "C2f", [128, true]], [-1, 1, "Conv", [256, 3, 2]], [-1, 6, "C2f", [256, true]], [-1, 1, "Conv", [512, 3, 2]], [-1, 6, "C2f", [512, true]], [-1, 1, "Conv", [1024, 3, 2]], [-1, 3, "C2f", [1024, true]], [-1, 1, "SPPF", [1024, 5]]], "head": [[-1, 1, "nn.Upsample", ["None", 2, "nearest"]], [[-1, 6], 1, "Concat", [1]], [-1, 3, "C2f", [512]], [-1, 1, "nn.Upsample", ["None", 2, "nearest"]], [[-1, 4], 1, "Concat", [1]], [-1, 3, "C2f", [256]], [-1, 1, "Conv", [256, 3, 2]], [[-1, 12], 1, "Concat", [1]], [-1, 3, "C2f", [512]], [-1, 1, "Conv", [512, 3, 2]], [[-1, 9], 1, "Concat", [1]], [-1, 3, "C2f", [1024]], [[15, 18, 21], 1, "Detect", ["nc"]]], "ch": 3}, "nc": 1, "args": {"model": "yolov8s.pt", "batch": 16, "imgsz": 640}, "ultralytics_version": "8.0.196", "model_type": "yolov8"}