MAJOR ACCURACY LOSS when EXPORTING tao unet model after retraining pruned model

[Morganh]

@david9xqqb

I trained a multiclass semantic segmentation unet model based on color images that has very poor performance.

So, let us make it clear and fix issues one by one.
I suggest you to train color images firstly.
See topic Problems encountered in training unet and inference unet - #27 by Morganh , it can get good accuracy.
You can leverage it.