Implement and Train U-NET From Scratch for Image Segmentation - PyTorch

preview_player
Показать описание
We will implement U-Net and train our implementation on the Carvana dataset! Want to support the channel? Hit that like button and subscribe! GitHub below ↓

GitHub Link of the Code

Notebook

UNet is introduced in the paper: "U-Net: Convolutional Networks for Biomedical Image Segmentation"

What should I implement next? Let me know in the comments!

00:00 Introduction
00:30 Paper Overview
03:25 Implement U-Net Parts
6:50 ViT Implement U-Net
11:50 Dataset Implementation
14:36 Dataset Preparation
16:11 Train Loop
19:06 Inference
21:35 Outro

Buy me a coffee! ☕️
Рекомендации по теме
Комментарии
Автор

Excellent tutorial! Thank you so much! May I ask, how much time did the training require on Google Colab?

sr__-n
Автор

Hi, I want to ask that the manual test and mask are the image we manually add from train and train_mask right?

an_nguyen
Автор

Can I use the same procedure with the CAMUS data set please reply

funnnnnnhubbb
welcome to shbcf.ru