2 ms·
This is pretty neat lol. I'm curious about what architecture you used, how many images and what kind of augmentation you used.
by newen 9y ago
This is pretty neat lol. I'm curious about what architecture you used, how many images and what kind of augmentation you used.
- eigenloss 9y agoTraining was a bit of a process. The architecture I use now is an 8-bit quantized mobilenet_v1_050_224 retrained on about 1000 versions of Loss.jpg (some quite similar, but not exact duplicates) and about 10,000 other images. All the images are 240x240 or thereabouts. I'll be publishing something soon on what it took to build that 10K image dataset - it was more challenging than I initially anticipated.