ssd512_vgg_voc.json

{
  "model": "ssd_vgg",
  "batchnorm": true,
  "input_info": {
    "sample_size":  [1, 3, 512, 512]
  },
  "pretrained": false,
  "basenet": "./vgg16_bn-6c64b313.pth", // taken from https://download.pytorch.org/models/vgg16_bn-6c64b313.pth
  "num_classes": 21,
  "dataset": "voc",
  "preprocessing": {
    "mean": [0.406, 0.456, 0.485],
    "std": [0.255, 0.224, 0.229],
    "normalize_coef": 255,
    "rgb": true
  },
  "batch_size": 160,
  "epochs": 740,
  "save_freq": 50,
  "optimizer": {
    "type": "Adam",
    "base_lr": 1e-4,
    "weight_decay": 5e-4,
    "schedule_type": "multistep",
    "steps": [1, 2, 3]
  },
  "ssd_params": {
    "steps": [8, 16, 32, 64, 128, 256, 512],
    "min_sizes": [35.84, 76.8, 153.6, 230.4, 307.2, 384.0, 460.8],
    "max_sizes": [76.8, 153.6, 230.4, 307.2, 384.0, 460.8, 537.6],
    "aspect_ratios": [[2], [2, 3], [2, 3], [2, 3], [2, 3], [2], [2]],
    "variance": [0.1, 0.1, 0.2, 0.2],
    "clip": false,
    "flip": true
  },
  "export_to_ir_via_onnx": true
}