search

snakers4 / yolov2-fish

My yolov2 tinkering for fish competition
0 stars 0 forks source link

shape of y_batch of the generator vs the output of the model doesn't match #1

Open sadakmed opened 5 years ago

sadakmed commented 5 years ago

Hii mate thank u for sharing this implementation it's really nice as a beginner.

during my cracking to your code, I didn't get it, normally u output the model as one-hot encoded but in y_batch of the generator is label encoded

this is get_generator from BatchGenerator class y_batch = np.zeros((self.config['BATCH_SIZE'], self.config['GRID_H'], self.config['GRID_W'], self.config['BOX'], 4+1+1)) # desired network output this is the output of the model

output = Reshape((GRID_H, GRID_W, BOX, 4 + 1 + CLASS))(x)

what do you think?

snakers4 commented 5 years ago

oh, this repo is such a mess better just take some clean implementation and train it from scratch )

sadakmed commented 5 years ago

@snakers4 could you please recommend me something

snakers4 commented 5 years ago

https://github.com/eriklindernoren/PyTorch-YOLOv3