Skip to content

Latest commit

 

History

History
32 lines (23 loc) · 1.25 KB

README.md

File metadata and controls

32 lines (23 loc) · 1.25 KB

PharyngeaLSeG

This project does 3D Multi Task Model.

Getting Started

These instructions will help you to get Pytorch Model instance.

Prerequisites

What things you need to install the software and how to install them:

  • Python (version 3.x)
  • Torch

Examples

# Get 3D Unet
from model.inception_resnet_v2.multi_task.multi_task_3d import InceptionResNetV2MultiTask3D
base_model = InceptionResNetV2MultiTask3D(input_shape=(1, 32, 512, 512),
                                         class_channel=2, seg_channels=2, validity_shape=(1, 8, 8, 8), 
                                         inject_class_channel=None,
                                         block_size=8, decode_init_channel=None,
                                         skip_connect=True, dropout_proba=0.05, norm="instance", act="relu6",
                                         class_act="softmax", seg_act="softmax", validity_act="sigmoid",
                                         get_seg=True, get_class=False, get_validity=False,
                                         use_class_head_simple=True, use_seg_pixelshuffle_only=False
                                         )
model_param_num = sum(p.numel() for p in base_model.parameters())
print(f"model_param_num = {model_param_num}")