Comments (1)
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)
- FileNotFoundError: TopDownMpiiDataset: file "/Lite-HRNet/configs/_base_/datasets/mpii.py" does not exist HOT 8
- Fuse layer problem HOT 1
- 计算量如何计算的?
- KeyError: 'LiteHRNet is already registered in models' HOT 10
- KeyError: "TopDown: 'TopDownSimpleHead is not in the models registry'" HOT 1
- HRNet or LiteHRNet? HOT 4
- Inference speed (frames per second) on Raspberry Pi?
- train mpii failed,MMDistributedDataParallel object has no attribute _sync_params
- ImportError: cannot import name 'build_posenet' from 'models' (unknown location) HOT 2
- Our new study on lightweight high resolution networks: Dite-HRNet HOT 1
- I got an issue about KeyError 'center',there is no keyword 'center' in result HOT 1
- Not able to download the test jsons?
- pretrained model problem
- Training custom model HOT 1
- Welcome update to OpenMMLab 2.0
- Why i got lower mIoU than HRNet-W16?
- cannot import name 'Config' from 'mmcv' HOT 3
- What is the model structure of small HRNet?
- ModuleNotFoundError: No module named 'mmcv.transforms' HOT 4
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
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.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from lite-hrnet.