FAIMED 3D

use fastai to quickly train fully three-dimensional models on radiological data

Classification

from faimed3d.all import *

Load data in various medical formats (DICOM, NIfTI, NRRD) or even videos as simple as in fastai.

d = pd.read_csv('../data/radiopaedia_cases.csv')
dls = ImageDataLoaders3D.from_df(d,
                                 item_tfms = Resize3D((20, 112, 112)),
                                 batch_tfms = aug_transforms_3d(), 
                                 bs = 2, val_bs = 2)

Faimed3d provides multiple model architectures, pretrained on the UCF101 dataset for action recoginiton, which can be used for transfer learning.

Model	3-fold accuracy	duration/epoch	model size
efficientnet b0	92.5 %	9M:35S	48.8 MB
efficientnet b1	90.1 %	13M:20S	80.5 MB
resnet 18	87.6 %	6M:57S	339.1 MB
resnet 50	94.8 %	12M:16S	561.2 MB
resnet 101	96.0 %	17M:20S	1,030 MB

# slow
learn = cnn_learner_3d(dls, efficientnet_b0)

# slow
learn.lr_find()

SuggestedLRs(lr_min=0.014454397559165954, lr_steep=6.309573450380412e-07)

Click here for a more in-depth classification example.

Segmentation

dls = SegmentationDataLoaders3D.from_df(d,
                                 item_tfms = Resize3D((20, 112, 112)),
                                 batch_tfms = aug_transforms_3d(), 
                                 bs = 2, val_bs = 2)

All models in faimed3d can be used as a backbone for U-Nets, even with pre-trained weights.

# slow
learn = unet_learner_3d(dls, efficientnet_b0, n_out = 2)

# slow 
learn.lr_find()

SuggestedLRs(lr_min=0.33113112449646, lr_steep=0.10000000149011612)

Click here for a more in-depth segmentation example.

Extension to fastai for volumetric medical data

Related tags

Overview

FAIMED 3D

Classification

Segmentation

Owner

Keno

Efficient and intelligent interactive segmentation annotation software

Node for thenewboston digital currency network.

Code release for "Detecting Twenty-thousand Classes using Image-level Supervision".

This repository contains the scripts for downloading and validating scripts for the documents

Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch.

Official implementation of Rethinking Graph Neural Architecture Search from Message-passing (CVPR2021)

Face Mask Detection system based on computer vision and deep learning using OpenCV and Tensorflow/Keras

This is a demo app to be used in the video streaming applications

A tiny, pedagogical neural network library with a pytorch-like API.

Generate text captions for images from their CLIP embeddings. Includes PyTorch model code and example training script.

Deep Learning Algorithms for Hedging with Frictions

Pytorch implementation of our paper under review — Lottery Jackpots Exist in Pre-trained Models

Python utility to generate filesystem content for Obsidian.

Compare outputs between layers written in Tensorflow and layers written in Pytorch

SANet: A Slice-Aware Network for Pulmonary Nodule Detection

This project deploys a yolo fastest model in the form of tflite on raspberry 3b+. The model is from another repository of mine called -Trash-Classification-Car

The Multi-Mission Maximum Likelihood framework (3ML)

Riemannian Convex Potential Maps

Code and data for "Broaden the Vision: Geo-Diverse Visual Commonsense Reasoning" (EMNLP 2021).

Intel® Nervana™ reference deep learning framework committed to best performance on all hardware