Feature: Prediction is now in batches

MarcoCaglia / fashion-image-classifier

The purpose of this repository is to train a binary image classifier, which can predict of a piece of clothing on an image is worn by a model, or if the piece is displayed by itself.

GNU General Public License v3.0

0 stars 0 forks source link

Feature: Prediction is now in batches #10

Closed MarcoCaglia closed 3 years ago

MarcoCaglia commented 3 years ago

This mitigates the consumption issue, but it does not solve it. Will probably need to work more with generators for this