baseline/export.json

{
  "name": "baseline",
  "source_checkpoint": "outputs/train/baseline/weights/best.pt",
  "engine": "outputs/tensorrt_fp16/baseline/model.engine",
  "input_shape": [
    1,
    3,
    640,
    640
  ],
  "output_shape_before_export": [
    1,
    10,
    8400
  ],
  "class_count": 6,
  "class_names": [
    "open",
    "short",
    "mousebite",
    "spur",
    "copper",
    "pin-hole"
  ],
  "precision": "fp16",
  "dynamic": false,
  "batch_size": 1,
  "parameters": 3012018,
  "counted_parameters": 3012018,
  "macs": 4073266400,
  "gmacs": 4.0732664,
  "flops_estimate": 8146532800,
  "gflops_estimate": 8.1465328,
  "gpu_name": "Tesla T4",
  "torch_version": "2.10.0+cu128",
  "cuda_version": "12.8",
  "tensorrt_version": "10.16.1.11",
  "ultralytics_version": "8.4.115"
}