This is all related to using activation: "sigmoid" If I remove that single line from the specs file, everything runs to completion (with the problems reported in my other two posts… )
What do you mean by default notebook ?? The isbi notebook? It took me a couple of hours to get it to run, and no. This doesn’t happen on the isbi notebook, but looking in deep into that example, the images are 320 by 320, which would not cause this issue. More wasted time…
I had trained a unet multiclass semantic segmentation model with color data of shape 704X1280…
All the processing completes, but the performance is very poor. Very very poor…
I have a C++ inference program that works, in the sense of taking the live feed data, normalizing, and pushing it into a cuda buffer by unfolding the rgb channels and doing NHWC to NCHW conversion, and it all works. With poor segmentation classification performance, but correct in regards to pixel placement.
Now, I am trying to improve performance by training a binary model to detect the one critical part of the image. After completing a full cycle of training, in addition to the major performance drop when exporting, the pixel placement is off, as if the columns and rows were swapped behind the scenes.
My estimation is that Nvidia unet has a completely different programing for grayscale-binary and for color-multi-class, and is completely different code… That’s why I think there is a major drop in performance when exporting, and sigmoid doesn’t work with images that are nor of square shape, and the inference is wrong in pixel placement…
There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one.
Thanks