-
Notifications
You must be signed in to change notification settings - Fork 518
/
Copy pathcoco2017_yolo_nas_pose_n.yaml
58 lines (50 loc) · 2.87 KB
/
coco2017_yolo_nas_pose_n.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
# YoloNASPose training on COCO2017 Dataset
# All YoloNASPose models trained in 640x640 resolution
#
# Instructions:
# 0. Have super-gradients installed (pip install super-gradients==3.3 or clone the repo and `pip install -e .`)
# 1. Make sure that the data is stored folder specified at `dataset_params.dataset_dir` (Default is /data/coco) or
# add "dataset_params.data_dir=<PATH-TO-DATASET>" at the end of the command below
# 2. Run the command to start the training:
# yolo_nas_pose_n: python -m super_gradients.train_from_recipe --config-name=coco2017_yolo_nas_pose_n
# yolo_nas_pose_s: python -m super_gradients.train_from_recipe --config-name=coco2017_yolo_nas_pose_s
# yolo_nas_pose_m: python -m super_gradients.train_from_recipe --config-name=coco2017_yolo_nas_pose_m
# yolo_nas_pose_l: python -m super_gradients.train_from_recipe --config-name=coco2017_yolo_nas_pose_l
#
# Training times and accuracies ([email protected] (COCO API, confidence 0.01, IoU threshold 0.7, test on original resolution):
# yolo_nas_pose_n: 93h on 8 NVIDIA GeForce RTX 3090, AP: 59.68 (val)
# yolo_nas_pose_s: 52h on 8 NVIDIA GeForce RTX 3090, AP: 64.15 (val)
# yolo_nas_pose_m: 57h on 8 NVIDIA GeForce RTX 3090, AP: 67.87 (val)
# yolo_nas_pose_l: 80h on 8 NVIDIA GeForce RTX 3090, AP: 68.24 (val)
#
# Offline evaluation using COCOEval for N variant:
# Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets= 20 ] = 0.597
# Average Precision (AP) @[ IoU=0.50 | area= all | maxDets= 20 ] = 0.832
# Average Precision (AP) @[ IoU=0.75 | area= all | maxDets= 20 ] = 0.657
# Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets= 20 ] = 0.540
# Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets= 20 ] = 0.685
# Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 20 ] = 0.656
# Average Recall (AR) @[ IoU=0.50 | area= all | maxDets= 20 ] = 0.879
# Average Recall (AR) @[ IoU=0.75 | area= all | maxDets= 20 ] = 0.713
# Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets= 20 ] = 0.598
# Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets= 20 ] = 0.738
defaults:
- training_hyperparams: coco2017_yolo_nas_pose_train_params
- dataset_params: coco_pose_estimation_yolo_nas_mosaic_dataset_params
- arch_params: yolo_nas_pose_n_arch_params
- checkpoint_params: default_checkpoint_params
- _self_
- variable_setup
architecture: yolo_nas_pose_n
multi_gpu: DDP
num_gpus: 8
experiment_suffix: ""
experiment_name: coco2017_${architecture}_${experiment_suffix}_${dataset_params.dataset_params_suffix}
arch_params:
num_classes: ${dataset_params.num_joints}
dataset_params:
mosaic_prob: 0.5
train_dataloader_params:
batch_size: 60
val_dataloader_params:
batch_size: 60