U-Net for GBM

Overview

My Final Year Project(FYP) In National University of Singapore(NUS)

You need

Pytorch(stable 1.9.1) 

Both cuda version and cpu version are OK

File Structure

📦FYP-U-Net
 ┣ 📂data
 ┃ ┣ 📂imgs
 ┃ ┃ ┣ 📌···.tif
 ┃ ┃ ┗ ···
 ┃ ┣ 📂masks
 ┃ ┃ ┣ 📌···_mask.tif
 ┃ ┃ ┗ ···
 ┃ ┣ 📂PredictImage 
 ┃ ┃ ┣ 📌0.tif
 ┃ ┃ ┣ 📌1.tif
 ┃ ┃ ┗ ···
 ┃ ┣ 📂SaveImage
 ┃ ┃ ┣ 📌0.tif
 ┃ ┃ ┣ 📌1.tif
 ┃ ┃ ┗ ···
 ┃ ┗ 📂Source
 ┃ ┃ ┣ 📂TCGA_CS_4941_19960909
 ┃ ┃ ┃ ┣ 📌TCGA_CS_4941_19960909_1.tif
 ┃ ┃ ┃ ┣ 📌TCGA_CS_4941_19960909_1_mask.tif 
 ┃ ┃ ┃ ┣ 📌TCGA_CS_4941_19960909_2.tif
 ┃ ┃ ┃ ┣ 📌TCGA_CS_4941_19960909_2_mask.tif 
 ┃ ┃ ┃ ┗ ···
 ┃ ┃ ┣ 📂TCGA_CS_4942_19970222
 ┃ ┃ ┗ ···
 ┣ 📂params
 ┃ ┗ 📜unet.pth
 ┣ 📓README,md
 ┣ 📄data.py
 ┣ 📄net.py
 ┣ 📄utils.py
 ┗ 📄train.py
  • 'data' dir contains the origin dataset in 'Source' dir. And the dataset can be download in Kaggle (https://www.kaggle.com/c/rsna-miccai-brain-tumor-radiogenomic-classification/). And also you can use different dataset.
  • 'imgs' contains images and 'masks' contains corresponding masks to images. Corresponding masks have a _mask suffix. More inforamtion you can check in kaggle.
  • 'SaveImage' is meant for store train results and 'PredictImage' is meant for store test results.
  • 'params' is meant for store model.

Quick Up

Run train.py

Change DataSet

  • Delte all images in data dir and its subdir.

  • Install dataset from kaggle or anything you like(PS. Corresponding masks must have a _mask suffix) into 'Source' dir

  • Run data.py

    python3 data.py
    

    Remember change the path. After this, you will get images and masks in imgs dir and masks dir.

  • Run train.py

    python3 train.py
    

    Remember change the path. And you can see the results in 'SaveImage' dir and 'PredictImage' dir.

Results

Segment Image

Pre-trained model

https://drive.google.com/file/d/1yyrITv7BQf9kDnP__g6Qa3_wUPD1c_i_/view?usp=sharing

Owner
PinkR1ver
Artist, go with the flow, stay up late
PinkR1ver
[PNAS2021] The neural architecture of language: Integrative modeling converges on predictive processing

The neural architecture of language: Integrative modeling converges on predictive processing Code accompanying the paper The neural architecture of la

Martin Schrimpf 36 Dec 01, 2022
Very deep VAEs in JAX/Flax

Very Deep VAEs in JAX/Flax Implementation of the experiments in the paper Very Deep VAEs Generalize Autoregressive Models and Can Outperform Them on I

Jamie Townsend 42 Dec 12, 2022
A Simple and Versatile Framework for Object Detection and Instance Recognition

SimpleDet - A Simple and Versatile Framework for Object Detection and Instance Recognition Major Features FP16 training for memory saving and up to 2.

TuSimple 3k Dec 12, 2022
"Segmenter: Transformer for Semantic Segmentation" reproduced via mmsegmentation

Segmenter-based-on-OpenMMLab "Segmenter: Transformer for Semantic Segmentation, arxiv 2105.05633." reproduced via mmsegmentation. We reproduce Segment

EricKani 22 Feb 24, 2022
The project page of paper: Architecture disentanglement for deep neural networks [ICCV 2021, oral]

This is the project page for the paper: Architecture Disentanglement for Deep Neural Networks, Jie Hu, Liujuan Cao, Tong Tong, Ye Qixiang, ShengChuan

Jie Hu 15 Aug 30, 2022
The end-to-end platform for building voice products at scale

Picovoice Made in Vancouver, Canada by Picovoice Picovoice is the end-to-end platform for building voice products on your terms. Unlike Alexa and Goog

Picovoice 318 Jan 07, 2023
REGTR: End-to-end Point Cloud Correspondences with Transformers

REGTR: End-to-end Point Cloud Correspondences with Transformers This repository contains the source code for REGTR. REGTR utilizes multiple transforme

Zi Jian Yew 108 Dec 17, 2022
Pytorch implementations of Bayes By Backprop, MC Dropout, SGLD, the Local Reparametrization Trick, KF-Laplace, SG-HMC and more

Bayesian Neural Networks Pytorch implementations for the following approximate inference methods: Bayes by Backprop Bayes by Backprop + Local Reparame

1.4k Jan 07, 2023
PyTorch Personal Trainer: My framework for deep learning experiments

Alex's PyTorch Personal Trainer (ptpt) (name subject to change) This repository contains my personal lightweight framework for deep learning projects

Alex McKinney 8 Jul 14, 2022
Problem-943.-ACMP - Problem 943. ACMP

Problem-943.-ACMP В "main.py" расположен вариант моего решения задачи 943 с серв

Konstantin Dyomshin 2 Aug 19, 2022
Code for paper Adaptively Aligned Image Captioning via Adaptive Attention Time

Adaptively Aligned Image Captioning via Adaptive Attention Time This repository includes the implementation for Adaptively Aligned Image Captioning vi

Lun Huang 45 Aug 27, 2022
Open-source Monocular Python HawkEye for Tennis

Tennis Tracking 🎾 Objectives Track the ball Detect court lines Detect the players To track the ball we used TrackNet - deep learning network for trac

ArtLabs 188 Jan 08, 2023
Official PyTorch code of DeepPanoContext: Panoramic 3D Scene Understanding with Holistic Scene Context Graph and Relation-based Optimization (ICCV 2021 Oral).

DeepPanoContext (DPC) [Project Page (with interactive results)][Paper] DeepPanoContext: Panoramic 3D Scene Understanding with Holistic Scene Context G

Cheng Zhang 66 Nov 16, 2022
Code for the bachelors-thesis flaky fault localization

Flaky_Fault_Localization Scripts for the Bachelors-Thesis: "Flaky Fault Localization" by Christian Kasberger. The thesis examines the usefulness of sp

Christian Kasberger 1 Oct 26, 2021
GNEE - GAT Neural Event Embeddings

GNEE - GAT Neural Event Embeddings This repository contains source code for the GNEE (GAT Neural Event Embeddings) method introduced in the paper: "Se

João Pedro Rodrigues Mattos 0 Sep 15, 2021
Self-supervised Multi-modal Hybrid Fusion Network for Brain Tumor Segmentation

JBHI-Pytorch This repository contains a reference implementation of the algorithms described in our paper "Self-supervised Multi-modal Hybrid Fusion N

FeiyiFANG 5 Dec 13, 2021
mbrl-lib is a toolbox for facilitating development of Model-Based Reinforcement Learning algorithms.

mbrl-lib is a toolbox for facilitating development of Model-Based Reinforcement Learning algorithms. It provides easily interchangeable modeling and planning components, and a set of utility function

Facebook Research 724 Jan 04, 2023
TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.

TensorFlow GNN This is an early (alpha) release to get community feedback. It's under active development and we may break API compatibility in the fut

889 Dec 30, 2022
Neural network pruning for finding a sparse computational model for controlling a biological motor task.

MothPruning Scientific Overview Originally inspired by biological nervous systems, deep neural networks (DNNs) are powerful computational tools for mo

Olivia Thomas 0 Dec 14, 2022
MixRNet(Using mixup as regularization and tuning hyper-parameters for ResNets)

MixRNet(Using mixup as regularization and tuning hyper-parameters for ResNets) Using mixup data augmentation as reguliraztion and tuning the hyper par

Bhanu 2 Jan 16, 2022