Sphere-Guided 3D Microwave Imaging Network as Point Cloud Shape with Offset-attention Moudule (TGRS)

This repository contains a Pytorch implementation of the paper:

Sphere-Guided 3D Microwave Imaging Network as Point Cloud Shape with Offset-attention Moudule.
Xueting Bi.

Dependencies

• Python 3.6
• CUDA 10.0.
• PyTorch. Codes are tested with version 1.2.0
• (Optional) TensorboardX for visualization of the training process.

Following is the suggested way to install these dependencies:

# Create a new conda environment
conda create -n wp python=3.6
conda activate wp

# Install pytorch (please refer to the commend in the official website)
conda install pytorch=1.2.0 torchvision cudatoolkit=10.0 -c pytorch -y

Usage

To train a model on point clouds sampled from 3D shapes:

python train.py

Log files and network parameters will be saved to log folder in default.

We provide various visulization function for shape interpolation, part interpolation, and so on.

python visual.py

Evaluation

Citation