所需安装包增加img2vec_pytorch.

4paradigm · Nov 29, 2021 · f50b013 · f50b013
1 parent c16c11b
commit f50b013
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -16,7 +16,8 @@
             'pandas',
             'sklearn',
             'tqdm',
-            'optuna'
+            'optuna',
+            'img2vec_pytorch'
         ],
         python_requires='>=3.6',
         # packages=[],