✨ Add PyTorch image classification example (#13134)

* ✨ add pytorch image classification example * 🔥 remove utils.py * 💄 fix flake8 style issues * 🔥 remove unnecessary line * ✨ limit dataset sizes * 📌 update reqs * 🎨 restructure - use datasets lib * 🎨 import transforms directly * 📝 add comments * 💄 style * 🔥 remove flag * 📌 update requirement warning * 📝 add vision README.md * 📝 update README.md * 📝 update README.md * 🎨 add image-classification tag to model card * 🚚 rename vision ➡️ image-classification * 📝 update image-classification README.md
2021-09-02 13:29:42 -06:00
parent 9bd5d97cdd
commit 76c4d8bf26
14 changed files with 529 additions and 0 deletions
--- a/tests/fixtures/tests_samples/cats_and_dogs/Cat/1.jpg
+++ b/tests/fixtures/tests_samples/cats_and_dogs/Cat/1.jpg
--- a/tests/fixtures/tests_samples/cats_and_dogs/Cat/2.jpg
+++ b/tests/fixtures/tests_samples/cats_and_dogs/Cat/2.jpg
--- a/tests/fixtures/tests_samples/cats_and_dogs/Cat/3.jpg
+++ b/tests/fixtures/tests_samples/cats_and_dogs/Cat/3.jpg
--- a/tests/fixtures/tests_samples/cats_and_dogs/Cat/4.jpg
+++ b/tests/fixtures/tests_samples/cats_and_dogs/Cat/4.jpg
--- a/tests/fixtures/tests_samples/cats_and_dogs/Cat/5.jpg
+++ b/tests/fixtures/tests_samples/cats_and_dogs/Cat/5.jpg
--- a/tests/fixtures/tests_samples/cats_and_dogs/Dog/1.jpg
+++ b/tests/fixtures/tests_samples/cats_and_dogs/Dog/1.jpg
--- a/tests/fixtures/tests_samples/cats_and_dogs/Dog/2.jpg
+++ b/tests/fixtures/tests_samples/cats_and_dogs/Dog/2.jpg
--- a/tests/fixtures/tests_samples/cats_and_dogs/Dog/3.jpg
+++ b/tests/fixtures/tests_samples/cats_and_dogs/Dog/3.jpg
--- a/tests/fixtures/tests_samples/cats_and_dogs/Dog/4.jpg
+++ b/tests/fixtures/tests_samples/cats_and_dogs/Dog/4.jpg
--- a/tests/fixtures/tests_samples/cats_and_dogs/Dog/5.jpg
+++ b/tests/fixtures/tests_samples/cats_and_dogs/Dog/5.jpg