| { |
| "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" |
| } |
|
|
