diff --git a/params/model/edgeyolo-visdrone.yaml b/params/model/edgeyolo-visdrone.yaml deleted file mode 100644 index 60b83f8..0000000 --- a/params/model/edgeyolo-visdrone.yaml +++ /dev/null @@ -1,140 +0,0 @@ -# parameters -nc: 10 # number of classes -depth_multiple: 1.0 # models depth multiple -width_multiple: 1.0 # layer channel multiple - -# anchor-box-free -anchors: - - [8, 8] # P3/8 - - [16, 16] # P4/16 - - [32, 32] # P5/32 - -# edgeyolo backbone -backbone: - # [from, number, module, args] - [[-1, 1, Conv, [32, 3, 1]], # 0 - - [-1, 1, Conv, [64, 3, 2]], # 1-P1/2 - [-1, 1, Conv, [64, 3, 1]], - - [-1, 1, Conv, [128, 3, 2]], # 3-P2/4 - [-1, 1, Conv, [64, 1, 1]], - [-2, 1, Conv, [64, 1, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [[-1, -3, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [256, 1, 1]], # 11 - - [-1, 1, MP, []], - [-1, 1, Conv, [128, 1, 1]], - [-3, 1, Conv, [128, 1, 1]], - [-1, 1, Conv, [128, 3, 2]], - [[-1, -3], 1, Concat, [1]], # 16-P3/8 - [-1, 1, Conv, [128, 1, 1]], - [-2, 1, Conv, [128, 1, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [[-1, -3, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [512, 1, 1]], # 24 - - [-1, 1, MP, []], - [-1, 1, Conv, [256, 1, 1]], - [-3, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [256, 3, 2]], - [[-1, -3], 1, Concat, [1]], # 29-P4/16 - [-1, 1, Conv, [256, 1, 1]], - [-2, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [[-1, -3, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [1024, 1, 1]], # 37 - - [-1, 1, MP, []], - [-1, 1, Conv, [512, 1, 1]], - [-3, 1, Conv, [512, 1, 1]], - [-1, 1, Conv, [512, 3, 2]], - [[-1, -3], 1, Concat, [1]], # 42-P5/32 - [-1, 1, Conv, [256, 1, 1]], - [-2, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [[-1, -3, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [1024, 1, 1]], # 50 - ] - -# edgeyolo head -head: - [[-1, 1, SPPCSPC, [512]], # 51 - - [-1, 1, Conv, [256, 1, 1]], - [-1, 1, nn.Upsample, [None, 2, 'nearest']], - [37, 1, Conv, [256, 1, 1]], # route backbone P4 - [[-1, -2], 1, Concat, [1]], - - [-1, 1, Conv, [256, 1, 1]], - [-2, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [256, 1, 1]], # 63 - - [-1, 1, Conv, [128, 1, 1]], - [-1, 1, nn.Upsample, [None, 2, 'nearest']], - [24, 1, Conv, [128, 1, 1]], # route backbone P3 - [[-1, -2], 1, Concat, [1]], - - [-1, 1, Conv, [128, 1, 1]], - [-2, 1, Conv, [128, 1, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [128, 1, 1]], # 75 - - [-1, 1, MP, []], - [-1, 1, Conv, [128, 1, 1]], - [-3, 1, Conv, [128, 1, 1]], - [-1, 1, Conv, [128, 3, 2]], - [[-1, -3, 63], 1, Concat, [1]], - - [-1, 1, Conv, [256, 1, 1]], - [-2, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [256, 1, 1]], # 88 - - [-1, 1, MP, []], - [-1, 1, Conv, [256, 1, 1]], - [-3, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [256, 3, 2]], - [[-1, -3, 51], 1, Concat, [1]], - - [-1, 1, Conv, [512, 1, 1]], - [-2, 1, Conv, [512, 1, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [512, 1, 1]], # 101 - - [75, 1, RepConv, [256, 3, 1]], # 102 - [88, 1, RepConv, [512, 3, 1]], # 103 - [101, 1, RepConv, [1024, 3, 1]], # 104 - - [[102,103,104], 1, YOLOXDetect, [nc, anchors, Conv]], - ] diff --git a/params/model/edgeyolo_coco.yaml b/params/model/edgeyolo_coco.yaml deleted file mode 100644 index c78791d..0000000 --- a/params/model/edgeyolo_coco.yaml +++ /dev/null @@ -1,141 +0,0 @@ -# parameters -nc: 80 # number of classes -depth_multiple: 1.0 # models depth multiple -width_multiple: 1.0 # layer channel multiple - -# anchor-box-free -anchors: - - [8, 8] # P3/8 - - [16, 16] # P4/16 - - [32, 32] # P5/32 - -# edgeyolo backbone -backbone: - # [from, number, module, args] - [[-1, 1, Conv, [32, 3, 1]], # 0 - - [-1, 1, Conv, [64, 3, 2]], # 1-P1/2 - [-1, 1, Conv, [64, 3, 1]], - - [-1, 1, Conv, [128, 3, 2]], # 3-P2/4 - [-1, 1, Conv, [64, 1, 1]], - [-2, 1, Conv, [64, 1, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [[-1, -3, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [256, 1, 1]], # 11 - - [-1, 1, MP, []], - [-1, 1, Conv, [128, 1, 1]], - [-3, 1, Conv, [128, 1, 1]], - [-1, 1, Conv, [128, 3, 2]], - [[-1, -3], 1, Concat, [1]], # 16-P3/8 - [-1, 1, Conv, [128, 1, 1]], - [-2, 1, Conv, [128, 1, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [[-1, -3, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [512, 1, 1]], # 24 - - [-1, 1, MP, []], - [-1, 1, Conv, [256, 1, 1]], - [-3, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [256, 3, 2]], - [[-1, -3], 1, Concat, [1]], # 29-P4/16 - [-1, 1, Conv, [256, 1, 1]], - [-2, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [[-1, -3, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [1024, 1, 1]], # 37 - - [-1, 1, MP, []], - [-1, 1, Conv, [512, 1, 1]], - [-3, 1, Conv, [512, 1, 1]], - [-1, 1, Conv, [512, 3, 2]], - [[-1, -3], 1, Concat, [1]], # 42-P5/32 - [-1, 1, Conv, [256, 1, 1]], - [-2, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [[-1, -3, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [1024, 1, 1]], # 50 - ] - -# edgeyolo head -head: - [[-1, 1, SPPCSPC, [512]], # 51 - - [-1, 1, Conv, [256, 1, 1]], - [-1, 1, nn.Upsample, [None, 2, 'nearest']], - [37, 1, Conv, [256, 1, 1]], # route backbone P4 - [[-1, -2], 1, Concat, [1]], - - [-1, 1, Conv, [256, 1, 1]], - [-2, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [256, 1, 1]], # 63 - - [-1, 1, Conv, [128, 1, 1]], - [-1, 1, nn.Upsample, [None, 2, 'nearest']], - [24, 1, Conv, [128, 1, 1]], # route backbone P3 - [[-1, -2], 1, Concat, [1]], - - [-1, 1, Conv, [128, 1, 1]], - [-2, 1, Conv, [128, 1, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [-1, 1, Conv, [64, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [128, 1, 1]], # 75 - - [-1, 1, MP, []], - [-1, 1, Conv, [128, 1, 1]], - [-3, 1, Conv, [128, 1, 1]], - [-1, 1, Conv, [128, 3, 2]], - [[-1, -3, 63], 1, Concat, [1]], - - [-1, 1, Conv, [256, 1, 1]], - [-2, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [-1, 1, Conv, [128, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [256, 1, 1]], # 88 - - [-1, 1, MP, []], - [-1, 1, Conv, [256, 1, 1]], - [-3, 1, Conv, [256, 1, 1]], - [-1, 1, Conv, [256, 3, 2]], - [[-1, -3, 51], 1, Concat, [1]], - - [-1, 1, Conv, [512, 1, 1]], - [-2, 1, Conv, [512, 1, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [-1, 1, Conv, [256, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [512, 1, 1]], # 101 - - [75, 1, RepConv, [256, 3, 1]], # 102 - [88, 1, RepConv, [512, 3, 1]], # 103 - [101, 1, RepConv, [1024, 3, 1]], # 104 - - [[102,103,104], 1, YOLOXDetect, [nc, anchors, Conv]], - ] - ] diff --git a/params/model/edgeyolo_coco_s.yaml b/params/model/edgeyolo_coco_s.yaml deleted file mode 100644 index 5d06ff2..0000000 --- a/params/model/edgeyolo_coco_s.yaml +++ /dev/null @@ -1,148 +0,0 @@ -# parameters -nc: 80 # number of classes -depth_multiple: 1.0 # models depth multiple -width_multiple: 1.0 # layer channel multiple - -# anchor-box-free -anchors: - - [8, 8] # P3/8 - - [16, 16] # P4/16 - - [32, 32] # P5/32 - -# edgeyolo backbone -backbone: - # [from, number, module, args] - [[-1, 1, Conv, [32, 3, 1]], # 0 - - [-1, 1, Conv, [64, 3, 2]], # 1-P1/2 - [-1, 1, Conv, [64, 3, 1]], - - [-1, 1, Conv, [80, 3, 2]], # 3-P2/4 - [-1, 1, Conv, [40, 1, 1]], - [-2, 1, Conv, [40, 1, 1]], - [-1, 1, Conv, [40, 3, 1]], - [-1, 1, Conv, [40, 3, 1]], - [-1, 1, Conv, [40, 3, 1]], - [-1, 1, Conv, [40, 3, 1]], - [[-1, -2], 1, Shortcut, [1]], - [[-4, -5], 1, Shortcut, [1]], - [[-1, -2, -7, -8], 1, Concat, [1]], - [-1, 1, Conv, [120, 1, 1]], # 13 - - [-1, 1, MP, []], - [-1, 1, Conv, [80, 1, 1]], - [-3, 1, Conv, [80, 1, 1]], - [-1, 1, Conv, [80, 3, 2]], - [[-1, -3], 1, Concat, [1]], # 18-P3/8 - [-1, 1, Conv, [80, 1, 1]], - [-2, 1, Conv, [80, 1, 1]], - [-1, 1, DWConv, [80, 3, 1]], - [-1, 1, Conv, [80, 3, 1]], - [-1, 1, DWConv, [80, 3, 1]], - [-1, 1, Conv, [80, 3, 1]], - [[-1, -2], 1, Shortcut, [1]], - [[-4, -5], 1, Shortcut, [1]], - [[-1, -2, -7, -8], 1, Concat, [1]], - [-1, 1, Conv, [240, 1, 1]], # 28 - - [-1, 1, MP, []], - [-1, 1, Conv, [160, 1, 1]], - [-3, 1, Conv, [160, 1, 1]], - [-1, 1, Conv, [160, 3, 2]], - [[-1, -3], 1, Concat, [1]], # 33-P4/16 - [-1, 1, Conv, [160, 1, 1]], - [-2, 1, Conv, [160, 1, 1]], - [-1, 1, DWConv, [160, 3, 1]], - [-1, 1, Conv, [160, 3, 1]], - [-1, 1, DWConv, [160, 3, 1]], - [-1, 1, Conv, [160, 3, 1]], - [[-1, -2], 1, Shortcut, [1]], - [[-4, -5], 1, Shortcut, [1]], - [[-1, -2, -7, -8], 1, Concat, [1]], - [-1, 1, Conv, [480, 1, 1]], # 43 - - [-1, 1, MP, []], - [-1, 1, Conv, [320, 1, 1]], - [-3, 1, Conv, [320, 1, 1]], - [-1, 1, Conv, [320, 3, 2]], - [[-1, -3], 1, Concat, [1]], # 48-P5/32 - [-1, 1, Conv, [160, 1, 1]], - [-2, 1, Conv, [160, 1, 1]], - [-1, 1, DWConv, [160, 3, 1]], - [-1, 1, Conv, [160, 3, 1]], - [-1, 1, DWConv, [160, 3, 1]], - [-1, 1, Conv, [160, 3, 1]], - [[-1, -2], 1, Shortcut, [1]], - [[-4, -5], 1, Shortcut, [1]], - [[-1, -2, -7, -8], 1, Concat, [1]], - [-1, 1, Conv, [480, 1, 1]], # 58 - ] - -# edgeyolo head -head: - [[-1, 1, SPPCSPC, [240]], # 59 - - [-1, 1, Conv, [120, 1, 1]], - [-1, 1, nn.Upsample, [None, 2, 'nearest']], - [43, 1, Conv, [120, 1, 1]], # route backbone P4 - [[-1, -2], 1, Concat, [1]], - - [-1, 1, Conv, [120, 1, 1]], - [-2, 1, Conv, [120, 1, 1]], - [-1, 1, DWConv, [60, 3, 1]], - [-1, 1, Conv, [60, 3, 1]], - [-1, 1, DWConv, [60, 3, 1]], - [-1, 1, Conv, [60, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [120, 1, 1]], # 71 - - [-1, 1, Conv, [60, 1, 1]], - [-1, 1, nn.Upsample, [None, 2, 'nearest']], - [28, 1, Conv, [60, 1, 1]], # route backbone P3 - [[-1, -2], 1, Concat, [1]], - - [-1, 1, Conv, [60, 1, 1]], - [-2, 1, Conv, [60, 1, 1]], - [-1, 1, DWConv, [30, 3, 1]], - [-1, 1, Conv, [30, 3, 1]], - [-1, 1, DWConv, [30, 3, 1]], - [-1, 1, Conv, [30, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [60, 1, 1]], # 83 - - [-1, 1, MP, []], - [-1, 1, Conv, [60, 1, 1]], - [-3, 1, Conv, [60, 1, 1]], - [-1, 1, Conv, [60, 3, 2]], - [[-1, -3, 71], 1, Concat, [1]], - - [-1, 1, Conv, [120, 1, 1]], - [-2, 1, Conv, [120, 1, 1]], - [-1, 1, DWConv, [60, 3, 1]], - [-1, 1, Conv, [60, 3, 1]], - [-1, 1, DWConv, [60, 3, 1]], - [-1, 1, Conv, [60, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [120, 1, 1]], # 96 - - [-1, 1, MP, []], - [-1, 1, Conv, [120, 1, 1]], - [-3, 1, Conv, [120, 1, 1]], - [-1, 1, Conv, [120, 3, 2]], - [[-1, -3, 59], 1, Concat, [1]], - - [-1, 1, Conv, [240, 1, 1]], - [-2, 1, Conv, [240, 1, 1]], - [-1, 1, DWConv, [120, 3, 1]], - [-1, 1, Conv, [120, 3, 1]], - [-1, 1, DWConv, [120, 3, 1]], - [-1, 1, Conv, [120, 3, 1]], - [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]], - [-1, 1, Conv, [240, 1, 1]], # 109 - - [83, 1, RepConv, [120, 3, 1]], # 110 - [96, 1, RepConv, [240, 3, 1]], # 111 - [109, 1, RepConv, [480, 3, 1]], # 112 - - [[110,111,112], 1, YOLOXDetect, [nc, anchors, Conv]], - ] diff --git a/params/train/train_settings.yaml b/params/train/train_settings.yaml deleted file mode 100644 index 67bd1cb..0000000 --- a/params/train/train_settings.yaml +++ /dev/null @@ -1,71 +0,0 @@ -# models & weights -model_cfg: "params/model/edgeyolo.yaml" -weights: "./train_output/edgeyolo/last.pth" # it contains models config, set null if not use it -use_cfg: false # force using cfg file data instead of cfg in weights - -# output -output_dir: "./train_output/edgeyolo" -save_checkpoint_for_each_epoch: true -log_file: "log.txt" - -# dataset -dataset_dir: "/dataset/coco2017" -train_dir: "train2017" -val_dir: "val2017" -train_anno_file: "annotations/instances_train2017.json" -val_anno_file: "annotations/instances_val2017.json" -class_names_file: "classes.txt" # this file is in dataset dir -use_segmentation: false # use segmentation labels to generate bbox instead of using bbox labels directly. - -# dataloader -batch_size_per_gpu: 8 -loader_num_workers: 4 -num_threads: 1 - -# device & data type -device: [0, 1, 2, 3] -fp16: false -cudnn_benchmark: false # set true if multiscale_range is set zero - -## train hyper-params -optimizer: "Adam" -max_epoch: 400 # 300 -close_mosaic_epochs: 15 - -# learning rate -lr_per_img: 0.00015625 -warmup_epochs: 5 -warmup_lr_ratio: 0.0 -final_lr_ratio: 0.01 # 0.05 (if epoch is 300) - -# training/dataset enhancement -# [cls_loss, conf_loss, iou_loss] -loss_use: ["bce", "bce", "giou"] # bce: BCE loss. bcf: Balanced Focal loss. hyb: HR loss, iou, c/g/siou is available - -input_size: [640, 640] -multiscale_range: 5 -weight_decay: 0.0005 -momentum: 0.9 -use_ema: true -enable_mixup: true -mixup_scale: [0.5, 1.5] -mosaic_scale: [0.1, 2.0] -flip_prob: 0.5 -mosaic_prob: 1 -mixup_prob: 1 -degrees: 10 - -# evaluate -eval_at_start: false -val_conf_thres: 0.01 -val_nms_thres: 0.65 -eval_only: false # running benchmark only for all weights in output_dir - -# show -print_interval: 10 - -# others -load_optimizer_params: true # load optimizer params when resume train, set false if there is an error. -train_backbone: true # set false if you only want to train head -train_start_layers: 51 # when not train backbone, train layers starts from this layer, see params/models/edgeyolo.yaml -force_start_epoch: -1 # set -1 to disable this option