Stack Overflow
Stack Overflow

Questions tagged [unet-neural-network]

190 questions
13
votes
3 answers

Tensorflow 2 throwing ValueError: as_list() is not defined on an unknown TensorShape

I'm trying to train a Unet model in Tensorflow 2.0 which takes as input an image and a segmentation mask, but I'm getting a ValueError : as_list() is not defined on an unknown TensorShape. The stack trace shows the problem occurs during…
CSharp
  • 1,396
  • 1
  • 18
  • 41
7
votes
3 answers

ModuleNotFoundError: No module named 'tensorflow.python.keras.engine.base_layer_v1

from keras import backend as K from tensorflow.keras.layers import MaxPooling2D,Conv2D,Input,Add,Flatten,AveragePooling2D,Dense,BatchNormalization,ZeroPadding2D,Activation from tensorflow.keras.models import Model def Dense_Layer(x,k): x =…
keshav swami
  • 551
  • 1
  • 6
  • 11
4
votes
1 answer

AttributeError: dataset object has no attribute 'c' FastAI

I am trying to train a ResNet based UNet for image segmentation. I have the location of images and mask images in a csv file, that's why I have created my own dataloader, which is as follows: X = list(df['input_img']) y =…
Sarvagya Gupta
  • 861
  • 3
  • 13
  • 28
4
votes
1 answer

Image segmentation - custom loss function in Keras

I am using a in Keras implemented U-Net (https://arxiv.org/pdf/1505.04597.pdf) to segment cell organelles in microscopy images. In order for my network to recognize multiple single objects that are separated by only 1 pixel, I want to use weight…
disputator1991
  • 165
  • 5
  • 13
3
votes
3 answers

Is there a way to print loss from Flux.train?

I'm trying to train a UNet in Julia with the help of Flux. Flux.train!(loss, Flux.params(model), train_data_loader, opt) batch_loss = loss(train_data, train_targets) where the loss is logitcrossentropy and train_data_loader…
h612
  • 544
  • 2
  • 11
3
votes
3 answers

How UNET is different from simple autoencoders?

UNET architecture is like first half encoder and second half decoder . There are different variations of autoencoders like sparse , variational etc. They all compress and decompress the data But the UNET is also same used for compressing and…
Hamza
  • 530
  • 5
  • 27
3
votes
1 answer

UNet for own RGB dataset for background removal in keras

I have 1000 cat images with background. I want to try UNet to remove background from those images. Any recommendations are welcome. Should I create annotations or other files like Json files? I have not found a proper way to do it.
icy-muddy
  • 33
  • 3
3
votes
1 answer

how to apply multi input image to Unet for segmentation?

I am a newbie in machine learning. Actually, I used my unet code for image segmentation using one input image slice (192x912) and one output mask image (192x192) My Unet code is contained several CNN layer and I usually used one input image…
Tom
  • 531
  • 4
  • 7
  • 19
3
votes
1 answer

Why could u-net mask image with smaller mask?

The input image size of u-net is 572*572, but the output mask size is 388*388. How could the image get masked with a smaller mask?
Ink
  • 845
  • 1
  • 13
  • 31
2
votes
2 answers

PointCloud to image fed to UNet: How to map predicted image back to 3D PointCloud?

I have a .las file and I performed the following operations: Convert PointCloud to RGB Image Convert PointCloud to GroundTruth Matrix. Crop Images and corresponding GroundTruth Matrix to fixed size 256x256 Train UNet (image and groundtuth…
h612
  • 544
  • 2
  • 11
2
votes
0 answers

How to save UNET predicted mask as an image to the disk in Pytorch?

AS a part of my Master's thesis, I have trained a UNET using Pytorch for detecting some objects in X-ray images. And to generate the predications, I have implemented the following function: def make_predictions(model, imagePath): # set model to…
Abdelrahman Youssef
  • 23
  • 4
2
votes
2 answers

Why is dilated convolution computationally efficient?

I have been studying UNet inspired architecture ENet and I think I follow the basic concepts. The ground-rock of efficiency of ENet is dilated convolution (apart other things). I understand the preserving spatial resolution, how it is computed and…
Michal Cicatka
  • 55
  • 1
  • 7
2
votes
0 answers

While training image segmentation network UNET- Validation loss is not getting decreased after some epochs, remains constant

I am training the UNET image segmentation network on brain tumor dataset from figshare, it is training well, training loss and training dice score both are changing accordingly or in the same tone with validation loss and validation dice score.…
Pankaj Kasar
  • 21
  • 5
2
votes
1 answer

Image Segmentation Tensorflow tutorials

In this tf tutorial, the U-net model has been divided into 2 parts, first contraction where they have used Mobilenet and it is not trainable. In second part, I'm not able to understand what all layers are being trained. As far as I could see, only…
JimmyFails
  • 113
  • 6
2
votes
3 answers

Is it possible to add the memory of 2 graphics cards together to run a larger neural network?

If I have a graphics card with 24 gb of ram, can I add a 2nd card, that is exactly the same, to double my memory to 48 gb? I want to run a large 3D-UNet but I am stopped due to the size of the volumes that I am passing. Will adding a second card…
rzaratx
  • 756
  • 3
  • 9
  • 29
1
2 3
…
12 13