Viewing a single comment thread. View all comments

Electronic-Art-2105 t1_irwzixb wrote

Reply to comment by AKavun in [P] Making attribute classification on an image of a clothing by AKavun

I see. In the tutorial, for each output, a 1-dimensional Dense layer with a sigmoid activation function is used, along with binary crossentropy as the loss function. You could exchange that by an n-dimensional Dense layer with softmax activation, along with categorical crossentropy. So the basic architecture can remain similar, you just have to adapt the outputs.

1

AKavun OP t1_irx83tz wrote

>I see. In the tutorial, for each output, a 1-dimensional Dense layer with a sigmoid activation function is used, along with binary crossentropy as the loss function. You could exchange that by an n-dimensional Dense layer with softmax activation, along with categorical crossentropy. So the basic architecture can remain similar, you just have to adapt the outputs.

I will first learn what these things mean, then I will get back to you. Thank you for your guidance.

2