Questions tagged [u-net]

For questions related to the U-net, a neural network proposed in "U-Net: Convolutional Networks for Biomedical Image Segmentation" (2015) by Olaf Ronneberger et al. for semantic segmentation.

26 questions with no upvoted or accepted answers
Filter by
Sorted by
Tagged with
4
votes
1answer
51 views

Validation accuracy higher than training accurarcy

I implemented the unet in tensorflow for the segmentation of MRI images of the thigh. I noticed I always get a higher validation accuracy by a small gap, independently of the initial split. One ...
4
votes
0answers
36 views

What are some neural network models that can use auxiliary info during training for image segmentation?

What are some deep learning models that can use supplementary information other than RGB channels for image segmentation? For example imagine a poorly shot image of a river (blue) that shows a gap, ...
3
votes
3answers
175 views

If I trained a model to perform semantic segmentation on images with only one object, would it also work on images with multiple objects?

I'm working on semantic segmentation tasks in the medical space using the U-Net. Let's say that I train a U-Net model on medical images with the goal of segmenting out, say, ligaments, from a medical ...
2
votes
0answers
39 views

How and why do state-of-the-art models in medical segmentation differ from general segmentation models?

I am just getting into medical image segmentation and have been able to understand the state-of-the-art architectures, like Double UNet, UNet++, and Multiresunet. What I haven't understood yet: Why ...
2
votes
0answers
61 views

Why do I get higher average dice accuracy for less data

I am working on image segmentation of MRI thigh images with deep learning (Unet). I noticed that I get a higher average dice accuracy over my predicted masks if I have less samples in the test data ...
2
votes
1answer
2k views

What are some good alternatives to U-Net for biomedical image segmentation?

Soon I will be working on biomedical image segmentation (microscopy images). There will be a small amount of data (a few dozens at best). Is there a neural network, that can compete with U-Net, in ...
2
votes
0answers
12 views

How do we stack two U-Nets to yield one final prediction?

I am trying to reproduce the model described in the paper DocUNet: Document Image Unwarping via A Stacked U-Net, i.e. stacking two U-Nets to yield one final prediction. The paper mentions that: ...
1
vote
0answers
21 views

Pixel values of segmap in multi-class semantic segmentation

I'm preparing a dataset for a multiclass semantic segmentation using U-Net like architecture. To be precise, I've got it ready but a question came to my mind. How does pixel values of a segmentation ...
1
vote
0answers
93 views

Hairstyle Virtual Try On

I want to help people with cancer who are under chemotherapy, and generally people who have lost their hair to Virtually Try-On Toupees/Wigs on their head. VTO must support both the frontal and side ...
1
vote
0answers
25 views

Could the data augmentation lead to the model learning features which corresponds to data augmented data and not to the real data?

I am trying to train a Unet network with Synthetic data to do binary segmentation due to the fact that is is not easy to collect real data. And there is something in the training process that I do not ...
1
vote
0answers
265 views

Getting bounding box/boundaries from segmentations in UNet Nuclei Segmentation

From my understanding, in a tissue where nuclei are present and need to be detected, we need to predict bounding boxes (either rectangular/circular or in the shape of the nucleus, i.e. as in instance ...
1
vote
0answers
18 views

Model output segmentation maps which are not full

I created a VGG based U-Net in order to perform image segmentation task on yeast cells images obtained by a microscope. There are a couple of problems with the data: There is inhomogeneity in the ...
1
vote
1answer
57 views

How to quickly change hand-drawn shapes to symmetrical polished shapes?

Given a hand-drawn shape, I'd like to generate the corresponding symmetrical polished shapes such as circle, rectangle, triangle, trapezoid, square, parallelogram, etc. A short video demonstration ...
1
vote
0answers
89 views

Is it necessary to label the background when generating the labelled dataset for semantic segmentation?

When I label images for semantic segmentation (using u-net, if that matters), is labeling the background (anything I am not interested in) necessary? Will it improve the network's performance?
1
vote
0answers
51 views

Suppress heatmap non-maxima in segmentation with UNet

I'm using U-Net for image segmentation. The model was trained with images that could contain up to 4 different classes. The train classes are never overlapping. The output of the UNet is a heatmap (...
1
vote
0answers
35 views

How to train image segmentation task with only one class?

Is there a neural network that has architecture optimizations for segmenting only one class (object and background)? I have tried U-net but it is not providing good enough results. I am wondering if ...
1
vote
0answers
44 views

Using U-NET for image semantic segmentation

I'm getting literally crazy trying to understand how U-NET works. Maybe it is very easy, but I'm stuck (and I have a terrible headache). So, I need your help. I'm going to segment MRI to find white ...
1
vote
0answers
7 views

Would models like U-Net be able to segment objects which has label based on its surrounding context?

Suppose that we want to segment a red blob from the image, normally you will have a class for this red blob e.g. 0. And every red blob you detected will have a class of 0. But, in my case, I want ...
1
vote
0answers
288 views

How can I use the bottleneck layer of the U-net to calculate the similarity between two images?

I would like to use the bottleneck layer of U-Net (the last layer of the encoder) to calculate the similarity between two images. For that, I have to somehow flatten the last layer of the encoder. In ...
0
votes
0answers
6 views

In a binary classification with CNNs features perform well by themselves, but performance degrades when features are used simultaneously for training

I am doing a binary classification with U-Net. When I use elevation as a feature to train my data the test set performs very well. When I use slope as a feature it performs really well. When I use ...
0
votes
0answers
10 views

How to get Attention Maps from Attention Gates in Attention UNET?

Contex I have Attention UNET for image segmentation. I use it for humans segmentation. Question Everything works fine. I want to get attention maps from my network, so I could see what my UNET is ...
0
votes
0answers
22 views

Why does my model only predict 1s or 0s in multiclass segmentation?

I am currently trying to train a UNet model on the dronedeploy segmentation dataset. My problem is that I only get outputs like this from my model: This means that in the output maps I only get 0s ...
0
votes
0answers
16 views

Why doesn't U-Net work with images different from the dataset?

I have implemented a U-Net, similar to this implementation, but for a different dataset, this one, to segment roads. It works fine using the test folder images, but, for example, when I pick a print ...
0
votes
1answer
36 views

Late Onset Augmentation

If I train a U-Net model for image segmentation (e.g. medical images) and start training until it converges and then add augmentation - can i expect similar results as if i train with augmentation ...
0
votes
1answer
37 views

Semantic segmentation failing in small instance detection

I performed semantic segmentation with U-net. My dataset consists of grayscale images of defects. After training the dataset for I got an metric accuracy of only 0.3 - 0.4 IOU. Eventhough it is merely ...
0
votes
1answer
320 views

Artifacts After pruning Unet CNN

I'm trying to make a dark image brighter using CNN-UNet architecture. When I train the network, I get the following results: When I cut the features in half for pruning, and do full train again, I ...