package files and tests

ManuelFay · Oct 30, 2021 · 4749c99 · 4749c99
1 parent ae16d7d
commit 4749c99
Show file tree

Hide file tree

Showing 20 changed files with 30 additions and 8 deletions.
diff --git a/LICENSE.txt b/LICENSE.txt
@@ -0,0 +1,17 @@
+MIT License
+Copyright (c) 2018 YOUR NAME
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
diff --git a/README.md b/README.md
@@ -156,4 +156,8 @@ Embedding files:
 - Integrate with local version control (git-lfs ?)
 
 Frontend:
-- Overall UX and design changes
+- Overall UX and design changes
+- Enable Image upload
+
+Deployment:
+- Dockerize and orchestrate containers (image uploader, storage, indexation pipeline, inference)
diff --git a/setup.cfg b/setup.cfg
@@ -0,0 +1,2 @@
+[metadata]
+description-file = README.md
diff --git a/setup.py b/setup.py
@@ -13,6 +13,7 @@
     packages=find_packages(include=["image_searcher", "image_searcher.*"]),
     install_requires=[
         "torch",
+        "numpy",
         "ftfy",
         "transformers",
         "Pillow",

diff --git a/test/__init__.py → tests/__init__.py b/test/__init__.py → tests/__init__.py
diff --git a/tests/test_data/bird.jpg b/tests/test_data/bird.jpg
diff --git a/tests/test_data/car.jpg b/tests/test_data/car.jpg
diff --git a/tests/test_data/car2.jpg b/tests/test_data/car2.jpg
diff --git a/tests/test_data/elephant.jpg b/tests/test_data/elephant.jpg
diff --git a/tests/test_data/friends.jpg b/tests/test_data/friends.jpg
diff --git a/tests/test_data/minesweeper.jpeg b/tests/test_data/minesweeper.jpeg
diff --git a/tests/test_data/profile.jpg b/tests/test_data/profile.jpg
diff --git a/tests/test_data/traffic.jpg b/tests/test_data/traffic.jpg
diff --git a/test/test_embedder/__init__.py → tests/test_embedder/__init__.py b/test/test_embedder/__init__.py → tests/test_embedder/__init__.py
diff --git a/test/test_embedder/test_embedder.py → tests/test_embedder/test_embedder.py b/test/test_embedder/test_embedder.py → tests/test_embedder/test_embedder.py
@@ -7,7 +7,7 @@
 
 class TestEmbedder(unittest.TestCase):
     def setUp(self):
-        self.loader = ImageLoader(image_dir_path="/home/manu/Downloads", traverse=True)
+        self.loader = ImageLoader(image_dir_path="./tests/test_data", traverse=True)
         self.embedder = ClipEmbedder()
 
     def test_image_embedder(self):

diff --git a/test/test_image_loader/__init__.py → tests/test_image_loader/__init__.py b/test/test_image_loader/__init__.py → tests/test_image_loader/__init__.py
diff --git a/test/test_image_loader/test_image_loader.py → tests/test_image_loader/test_image_loader.py b/test/test_image_loader/test_image_loader.py → tests/test_image_loader/test_image_loader.py
@@ -5,9 +5,9 @@
 
 class TestLoader(unittest.TestCase):
     def setUp(self):
-        self.loader = ImageLoader(image_dir_path="/home/manu/Downloads")
+        self.loader = ImageLoader(image_dir_path="./tests/test_data")
 
     def test_loader(self):
         images = self.loader.search_tree()
-        self.assertTrue(len(images)>10)
+        self.assertTrue(len(images) > 5)
         self.assertIsInstance(images, list)
diff --git a/test/test_search/__init__.py → tests/test_search/__init__.py b/test/test_search/__init__.py → tests/test_search/__init__.py
diff --git a/test/test_search/test_search.py → tests/test_search/test_search.py b/test/test_search/test_search.py → tests/test_search/test_search.py
@@ -8,8 +8,7 @@
 class TestSearch(unittest.TestCase):
     def setUp(self):
         logging.basicConfig(filename=None, level=logging.INFO)
-        self.image_dir_path = "/home/manu/perso/ImageSearcher/data/"
-        # self.image_dir_path = "/home/manu/Downloads/"
+        self.image_dir_path = "./tests/test_data"
         self.tearDown()
 
     def tearDown(self) -> None:

diff --git a/test/test_search/test_search_faces.py → tests/test_search/test_search_faces.py b/test/test_search/test_search_faces.py → tests/test_search/test_search_faces.py
@@ -8,8 +8,7 @@
 class TestSearch(unittest.TestCase):
     def setUp(self):
         logging.basicConfig(filename=None, level=logging.INFO)
-        self.image_dir_path = "/home/manu/perso/ImageSearcher/data/"
-        # self.image_dir_path = "/home/manu/Downloads/"
+        self.image_dir_path = "./tests/test_data"
         self.tearDown()
 
     def tearDown(self) -> None: