Submitted by viertys t3_125xdrq in MachineLearning
azorsenpai t1_je6hjpu wrote
Is there any reason you're really restraining to a Unet based model ? I'd recommend testing different architectures such as DeepLab V3 or FPN and see whether stuff improves. If it doesn't I'd recommend looking to your data and the quality of the ground truth as with only 100 data points you should be very much limited by the information contained in your data.
If the data is clean I'd recommend using some kind of ensemble method, this might be overkill, especially with heavy models but having multiple models with random initializations infer on a same input generally gives a few more points of accuracy/dice so if you really need it , this is an option.
viertys OP t1_je6peyv wrote
I started with U-Net, but I'm open to other architectures. I will try out DeepLab V3, thank you!
I believe the data is generally clean. Sadly, I can't get more data as all the datasets used in the research papers that I've read are private.
Viewing a single comment thread. View all comments