NVIDIA / ai-assisted-annotation-client

Client side integration example source code and libraries for AI-Assisted Annotation SDK
Other
307 stars 64 forks source link

Fix Dependencies and API signature for AIAA-PYTHON-Client #30

Closed SachidanandAlle closed 5 years ago

SachidanandAlle commented 5 years ago

1) Only have standard dependencies (SimpleITK and numpy) 2) Simplify API signatures for py-client (same as c++ client) 3) Fix the Test Config 4) Update the readme (add sample code) 5) Verified to support on windows/ubuntu for both python 2.7/3.x

This will help for other use-cases like 3D-Slicer to directly use py-client API and implement the required extensions

Example

import client_api
client = client_api.AIAAClient(server_ip='0.0.0.0', server_port=5000)

# List Models
models = client.model_list(label='spleen')

# Run Segmentation
client.segmentation(
  model='segmentation_ct_spleen', 
  image_in='input3D.nii.gz',
  image_out='seg_mask3D.nii.gz')

# Run Annotation
client.dextr3d(
   model='annotation_ct_spleen',
   point_set=[[52,181,126],[137,152,171],[97,113,148],[78,227,115],[72,178,80],[97,175,188]],
   image_in='input3D.nii.gz',
   image_out='ann_mask3D.nii.gz')

@holgerroth @anfeng