Files
tensordex/LabelsGenerator.py
T
Lucas Oskorep fbac5d519a Adding in new trained models, as well as new tools for reviewing the results.
Added in testing flow for testing our unfininshed/finished models.

Also adding a test dataset with one picture of every pokemon in the game.
2019-04-30 13:00:24 -05:00

5 lines
160 B
Python

import pandas as pd
import os
pd.DataFrame(sorted([f.name for f in os.scandir("./data/train") if f.is_dir()])).to_csv("labels.txt", index=False, header=False)