Skip to content

Commit

Permalink
initial medical image support
Browse files Browse the repository at this point in the history
  • Loading branch information
ad12 committed Mar 12, 2023
1 parent 4ee7f36 commit 3aa71ff
Show file tree
Hide file tree
Showing 4 changed files with 35 additions and 1 deletion.
17 changes: 17 additions & 0 deletions meerkat/columns/deferred/file.py
Expand Up @@ -17,6 +17,7 @@
from PIL import Image

import meerkat.tools.docs as docs
from meerkat import env
from meerkat.block.deferred_block import DeferredOp
from meerkat.columns.abstract import Column
from meerkat.columns.deferred.base import DeferredCell, DeferredColumn
Expand All @@ -30,6 +31,11 @@
from meerkat.interactive.formatter.base import FormatterGroup
from meerkat.interactive.formatter.image import DeferredImageFormatterGroup

if env.is_package_installed("voxel"):
import voxel
else:
voxel = None

logger = logging.getLogger(__name__)


Expand Down Expand Up @@ -485,6 +491,11 @@ def load_text(path: Union[str, io.BytesIO]):
return f.read()


# TODO: Add @requires decorator when available.
def load_medimg(path: Union[str, io.BytesIO]):
return voxel.read(path)


FILE_TYPES = {
"image": {
"loader": load_image,
Expand Down Expand Up @@ -512,6 +523,12 @@ def load_text(path: Union[str, io.BytesIO]):
"formatters": CodeFormatterGroup,
"exts": [".py", ".js", ".css", ".json", ".java", ".cpp", ".c", ".h", ".hpp"],
},
"medimg": {
"loader": load_medimg,
"formatters": DeferredImageFormatterGroup,
"exts": [".dcm", ".nii", ".nii.gz"],
"defer": False,
},
}


Expand Down
@@ -0,0 +1,15 @@
<script lang="ts">
import type { Endpoint } from "$lib/utils/types";
export let data: string;
export let classes: string = '';
export let on_scroll: Endpoint = null;
export let on_scroll_axis_change: Endpoint = null;
</script>


<div>
<img class={classes} src={data} alt="A Meerkat cell." />
</div>
@@ -0,0 +1,2 @@
class NdImage:
pass
2 changes: 1 addition & 1 deletion setup.py
Expand Up @@ -102,7 +102,7 @@
"vision": ["torchvision>=0.9.0", "opencv-python", "Pillow"],
"audio": ["torchaudio"],
"medimg": [
"dosma>=0.0.13",
"pyvoxel",
"kaggle",
"google-cloud-storage",
"google-cloud-bigquery[bqstorage,pandas]",
Expand Down

0 comments on commit 3aa71ff

Please sign in to comment.