Code Monkey home page Code Monkey logo

Comments (1)

JWSunny avatar JWSunny commented on June 12, 2024

config.yaml 内容如下:

AUTO_RESUME: true
CUDNN:
BENCHMARK: true
DETERMINISTIC: false
ENABLED: true
DATA_DIR: ''
GPUS: (0,1)
OUTPUT_DIR: 'LiteHRNet_w18_output'
LOG_DIR: 'LiteHRNet_w18_log'
WORKERS: 8
PRINT_FREQ: 300

DATASET:
COLOR_RGB: false
DATASET: 'coco'
ROOT: '/mnt/share/COCO/'
TEST_SET: 'val2017'
TRAIN_SET: 'train2017'
NUM_JOINTS_HALF_BODY: 8
PROB_HALF_BODY: 0.3
FLIP: true
ROT_FACTOR: 45
SCALE_FACTOR: 0.35
MODEL:
NAME: 'LiteHRNet'
MODEL_FILE: ''
INIT_WEIGHTS: true
IMAGE_SIZE:

  • 256
  • 256
    HEATMAP_SIZE:
  • 64
  • 64
    SIGMA: 2
    NUM_JOINTS: 17
    BASE_CHANNEL: 40
    TARGET_TYPE: 'gaussian'
    RATIO: 0.5
    NUM_STAGES: 3
    STAGE_REPEATS:
  • 2
  • 4
  • 2
    STAGE_BRANCHES:
  • 2
  • 3
  • 4
    STAGE_BLOCKS:
  • 2
  • 2
  • 2
    MODULE_TYPE:
  • 'LITE'
  • 'LITE'
  • 'LITE'
    WITH_FUSE:
  • True
  • True
  • True
    REDUCE_RATIOS:
  • 8
  • 8
  • 8
    WITH_HEAD: True

LOSS:
USE_TARGET_WEIGHT: true
TRAIN:
BATCH_SIZE_PER_GPU: 32
SHUFFLE: true
BEGIN_EPOCH: 0
END_EPOCH: 210
OPTIMIZER: 'adam'
LR: 0.002
LR_FACTOR: 0.1
LR_STEP:

  • 160
  • 190
    WD: 0.0001
    GAMMA1: 0.99
    GAMMA2: 0.0
    MOMENTUM: 0.9
    NESTEROV: false
    TEST:
    BATCH_SIZE_PER_GPU: 32
    COCO_BBOX_FILE: '/mnt/share/COCO/person_detection_results/COCO_val2017_detections_AP_H_56_person.json'
    BBOX_THRE: 1.0
    IMAGE_THRE: 0.0
    IN_VIS_THRE: 0.2
    MODEL_FILE: 'LiteHRNet_w18_output/coco/LiteHRNet/LiteHRNet_w18_256x256_coco_better_lr1e-3/model_best.pth'
    NMS_THRE: 1.0
    OKS_THRE: 0.9
    FLIP_TEST: true
    POST_PROCESS: true
    BLUR_KERNEL: 11
    USE_GT_BBOX: true
    DEBUG:
    DEBUG: true
    SAVE_BATCH_IMAGES_GT: true
    SAVE_BATCH_IMAGES_PRED: true
    SAVE_HEATMAPS_GT: false
    SAVE_HEATMAPS_PRED: false

from lite-hrnet.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.