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
2021:"Bridging Global Context Interactions for High-Fidelity Image Completion"

TFill arXiv | Project This repository implements the training, testing and editing tools for "Bridging Global Context Interactions for High-Fidelity I

Chuanxia Zheng 111 Jan 08, 2023
Automatic Attendance marker for LMS Practice School Division, BITS Pilani

LMS Attendance Marker Automatic script for lazy people to mark attendance on LMS for Practice School 1. Setup Add your LMS credentials and time slot t

Nihar Bansal 3 Jun 12, 2021
MPRNet-Cloud-removal: Progressive cloud removal

MPRNet-Cloud-removal Progressive cloud removal Requirements 1.Pytorch = 1.0 2.Python 3 3.NVIDIA GPU + CUDA 9.0 4.Tensorboard Installation 1.Clone the

Semi 95 Dec 18, 2022
Author Disambiguation using Knowledge Graph Embeddings with Literals

Author Name Disambiguation with Knowledge Graph Embeddings using Literals This is the repository for the master thesis project on Knowledge Graph Embe

12 Oct 19, 2022
Face and other object detection using OpenCV and ML Yolo

Object-and-Face-Detection-Using-Yolo- Opencv and YOLO object and face detection is implemented. You only look once (YOLO) is a state-of-the-art, real-

Happy N. Monday 3 Feb 15, 2022
Do Smart Glasses Dream of Sentimental Visions? Deep Emotionship Analysis for Eyewear Devices

EMOShip This repository contains the EMO-Film dataset described in the paper "Do Smart Glasses Dream of Sentimental Visions? Deep Emotionship Analysis

1 Nov 18, 2022
Pytorch implementation of the paper DocEnTr: An End-to-End Document Image Enhancement Transformer.

DocEnTR Description Pytorch implementation of the paper DocEnTr: An End-to-End Document Image Enhancement Transformer. This model is implemented on to

Mohamed Ali Souibgui 74 Jan 07, 2023
Defending graph neural networks against adversarial attacks (NeurIPS 2020)

GNNGuard: Defending Graph Neural Networks against Adversarial Attacks Authors: Xiang Zhang ( Zitnik Lab @ Harvard 44 Dec 07, 2022

Brax is a differentiable physics engine that simulates environments made up of rigid bodies, joints, and actuators

Brax is a differentiable physics engine that simulates environments made up of rigid bodies, joints, and actuators. It's also a suite of learning algorithms to train agents to operate in these enviro

Google 1.5k Jan 02, 2023
Implementation for "Seamless Manga Inpainting with Semantics Awareness" (SIGGRAPH 2021 issue)

Seamless Manga Inpainting with Semantics Awareness [SIGGRAPH 2021](To appear) | Project Website | BibTex Introduction: Manga inpainting fills up the d

101 Jan 01, 2023
LocUNet is a deep learning method to localize a UE based solely on the reported signal strengths from a set of BSs.

LocUNet LocUNet is a deep learning method to localize a UE based solely on the reported signal strengths from a set of BSs. The method utilizes accura

4 Oct 05, 2022
Sequence-to-Sequence learning using PyTorch

Seq2Seq in PyTorch This is a complete suite for training sequence-to-sequence models in PyTorch. It consists of several models and code to both train

Elad Hoffer 514 Nov 17, 2022
Voice of Pajlada with model and weights.

Pajlada TTS Stripped down version of ForwardTacotron (https://github.com/as-ideas/ForwardTacotron) with pretrained weights for Pajlada's (https://gith

6 Sep 03, 2021
.NET bindings for the Pytorch engine

TorchSharp TorchSharp is a .NET library that provides access to the library that powers PyTorch. It is a work in progress, but already provides a .NET

Matteo Interlandi 17 Aug 30, 2021
KwaiRec: A Fully-observed Dataset for Recommender Systems (Density: Almost 100%)

KuaiRec: A Fully-observed Dataset for Recommender Systems (Density: Almost 100%) KuaiRec is a real-world dataset collected from the recommendation log

Chongming GAO (高崇铭) 70 Dec 28, 2022
Tensor-Based Quantum Machine Learning

TensorLy_Quantum TensorLy-Quantum is a Python library for Tensor-Based Quantum Machine Learning that builds on top of TensorLy and PyTorch. Website: h

TensorLy 85 Dec 03, 2022
SGPT: Multi-billion parameter models for semantic search

SGPT: Multi-billion parameter models for semantic search This repository contains code, results and pre-trained models for the paper SGPT: Multi-billi

Niklas Muennighoff 182 Dec 29, 2022
Implementation of the paper NAST: Non-Autoregressive Spatial-Temporal Transformer for Time Series Forecasting.

Non-AR Spatial-Temporal Transformer Introduction Implementation of the paper NAST: Non-Autoregressive Spatial-Temporal Transformer for Time Series For

Chen Kai 66 Nov 28, 2022
A Comprehensive Analysis of Weakly-Supervised Semantic Segmentation in Different Image Domains (IJCV submission)

wsss-analysis The code of: A Comprehensive Analysis of Weakly-Supervised Semantic Segmentation in Different Image Domains, arXiv pre-print 2019 paper.

Lyndon Chan 48 Dec 18, 2022
Reference models and tools for Cloud TPUs.

Cloud TPUs This repository is a collection of reference models and tools used with Cloud TPUs. The fastest way to get started training a model on a Cl

5k Jan 05, 2023