0
$\begingroup$

I was working with segmentation using u-net and MobileNet.

While I trained with input size 256*256 it had an output with Val loss: 0.044 (In this time dense layer was 256, 128, 64, 32 with a learning rate of 0.0001).

But I needed to modify input for deployment to 128*128(see below code). But then the val_loss increased drastically around 0.052 which is not acceptable.


EPOCHS = 50
BATCH_SIZE = 8
LR = 0.0001
BUFFER_SIZE = 1000
VAL_SUBSPLITS = 5
SPLIT=0.3
SMOOTH = 1e-15

IMG_WIDTH=128
IMG_HEIGHT=128

# Expansive path
decode_1 = decoder_block(bridge_layer, encode_4, 256) 
decode_2 = decoder_block(decode_1, encode_3, 128)       
decode_3 = decoder_block(decode_2, encode_2, 64)             
decode_4 = decoder_block(decode_3, encode_1, 32)                    

What should I take care of?

$\endgroup$

0

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.