Skip to content

Latest commit

 

History

History
4 lines (3 loc) · 354 Bytes

File metadata and controls

4 lines (3 loc) · 354 Bytes

album-artwork-classification

Deep learning model to classify album genres by their artwork.

Master branch contains the multi-label model, while the "single-label classifier" branch contains the single-label model. To obtain the dataset, simply run build_dataset.py, which will initiate the fetch requests to download all of the album images required.