3D CT Scan Segmentation With Occupancy Network

This repo implements a 3D superresolution segmentation task for an airport baggage dataset.

Our final paper can be found here.

Model Architecture

To solve the problem of binary classification an Occupancy Network is utilize. The occupancy network is implemented with and without concatenation of the latent tensor from the encoding to the decoding path. Additionally conditional batch normalization in the decoding path can be utilized.

Original occupancy network with conditional batch normalization and without concatenation Occupancy network without conditional batch normalization and with concatenation

Usage

The occupancy network can be trained and/or tested, on the airport baggage dataset, by simply executing the main file.

python main.py

The following arguments can be passed to the main.py script.

Results