File size: 1,876 Bytes
436df5c | 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 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 | {
"model_type": "garment-mask-generation",
"task": "image-segmentation",
"pipeline_tag": "image-segmentation",
"architectures": ["GarmentMaskPipeline"],
"model_config": {
"device": "cuda:0",
"output_height": 1024,
"process_size": 512,
"use_convex_hull": true,
"schp_batch_size": 12,
"save_images": false
},
"checkpoint_paths": {
"densepose_ckpt": "chkpt/DensePose",
"schp_atr_ckpt": "chkpt/SCHP/exp-schp-201908301523-atr.pth",
"schp_lip_ckpt": "chkpt/SCHP/exp-schp-201908261155-lip.pth"
},
"mask_strategies": {
"allowed_strategies": ["ellipse", "box", "poly"],
"default_probabilities": {
"ellipse": 0.50,
"box": 0.30,
"poly": 0.20
}
},
"garment_types": {
"supported_types": ["upper", "lower", "dress", "upper_body", "lower_body", "full_body", "full"],
"type_mapping": {
"upper_body": "upper",
"lower_body": "lower",
"full_body": "dress",
"full": "dress"
}
},
"image_processing": {
"default_margin_ratio": 0.1,
"default_target_ratio": [4, 3],
"default_densepose_height": 1024,
"default_densepose_width": 768,
"default_final_height": 1024,
"default_final_width": 768
},
"models_info": {
"densepose": {
"architecture": "R_50_FPN_s1x",
"framework": "detectron2",
"task": "dense_pose_estimation",
"num_body_parts": 24
},
"schp_atr": {
"architecture": "ResNet101",
"framework": "pytorch",
"task": "human_parsing",
"dataset": "ATR",
"num_classes": 18,
"input_size": [512, 512]
},
"schp_lip": {
"architecture": "ResNet101",
"framework": "pytorch",
"task": "human_parsing",
"dataset": "LIP",
"num_classes": 20,
"input_size": [473, 473]
}
},
"version": "1.0.0",
"transformers_version": "4.36.0"
}
|