eizen-nsga 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

eizen_nsga/__init__.py

@@ -8,7 +8,7 @@ Usage:
     result = model.predict("image.jpg")
 """
 
-__version__ = "1.0.0"
+__version__ = "1.0.3"
 
 from .model import NASModel
 

eizen_nsga/model.py

@@ -61,6 +61,15 @@ class NASModel:
         self.model_category = self.config.get('modelCategory', 'SOTA')
         self.task = self.config.get('task', 'classification')
 
+        # Set up class names (like YOLO's model.names)
+        class_names = self.config.get('class_names', None)
+        if class_names and isinstance(class_names, list):
+            self.names = {i: name for i, name in enumerate(class_names)}
+        else:
+            # Fallback to indices if no names available
+            num_classes = self.config.get('num_classes', 0)
+            self.names = {i: str(i) for i in range(num_classes)}
+
         # Set device
         if device is None:
             device = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -139,6 +148,7 @@ class NASModel:
             'num_cells': r'num_cells[:\s]*(\d+)',
             'num_classes': r'num_classes[:\s]*(\d+)',
             'image_size': r'image_size[:\s]*(\d+)',
+            'class_names': r'class_names[:\s]*(.+)',
         }
 
         for key, pattern in patterns.items():

{eizen_nsga-1.0.1.dist-info → eizen_nsga-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: eizen-nsga
-Version: 1.0.1
+Version: 1.0.3
 Summary: Simple inference package for NSGA-Net trained models
 Home-page: https://github.com/eizen-ai/nsga-net
 Author: Eizen.ai Team
@@ -21,28 +21,24 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: torch>=1.9.0
-Requires-Dist: torchvision>=0.10.0
-Requires-Dist: numpy>=1.19.0
-Requires-Dist: pillow>=8.0.0
-Provides-Extra: sota
-Requires-Dist: ultralytics>=8.0.0; extra == "sota"
+Requires-Dist: numpy >=1.19.0
+Requires-Dist: pillow >=8.0.0
+Requires-Dist: torch >=1.9.0
+Requires-Dist: torchvision >=0.10.0
+Provides-Extra: all
+Requires-Dist: pandas >=1.0.0 ; extra == 'all'
+Requires-Dist: scikit-learn >=0.24.0 ; extra == 'all'
+Requires-Dist: tokenizers >=0.10.0 ; extra == 'all'
+Requires-Dist: transformers >=4.0.0 ; extra == 'all'
+Requires-Dist: ultralytics >=8.0.0 ; extra == 'all'
 Provides-Extra: nn
-Requires-Dist: pandas>=1.0.0; extra == "nn"
-Requires-Dist: scikit-learn>=0.24.0; extra == "nn"
+Requires-Dist: pandas >=1.0.0 ; extra == 'nn'
+Requires-Dist: scikit-learn >=0.24.0 ; extra == 'nn'
+Provides-Extra: sota
+Requires-Dist: ultralytics >=8.0.0 ; extra == 'sota'
 Provides-Extra: transformer
-Requires-Dist: transformers>=4.0.0; extra == "transformer"
-Requires-Dist: tokenizers>=0.10.0; extra == "transformer"
-Provides-Extra: all
-Requires-Dist: ultralytics>=8.0.0; extra == "all"
-Requires-Dist: pandas>=1.0.0; extra == "all"
-Requires-Dist: scikit-learn>=0.24.0; extra == "all"
-Requires-Dist: transformers>=4.0.0; extra == "all"
-Requires-Dist: tokenizers>=0.10.0; extra == "all"
-Dynamic: author
-Dynamic: home-page
-Dynamic: license-file
-Dynamic: requires-python
+Requires-Dist: tokenizers >=0.10.0 ; extra == 'transformer'
+Requires-Dist: transformers >=4.0.0 ; extra == 'transformer'
 
 # eizen-nsga
 

{eizen_nsga-1.0.1.dist-info → eizen_nsga-1.0.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
-eizen_nsga/__init__.py,sha256=SYzEu-7pzFy8Ul-bkjnNzuRQtvVQcfbTLBG1Dapun5g,291
-eizen_nsga/model.py,sha256=bIasvlnrVmaXnaYza7J9dMDoFIK-X1LC2PcPNIzdaBI,13224
+eizen_nsga/__init__.py,sha256=T3XKMNN2_ewjqX_imQH6q9TFafBTIyF0pkgj1OxPSX4,291
+eizen_nsga/model.py,sha256=NC1gyGdh8uwfzpQ8pYPLHd3pLOvFxtGauD1nZN6zgm8,13728
 eizen_nsga/sota/__init__.py,sha256=3hyX_CR86TvV2RYQaES0FhZdtklUBu3DB7HAv1Z0yGo,525
 eizen_nsga/sota/detection_heads.py,sha256=Tx7qgqc0_BBtaIenxNHg_dIwbDHyOQz3QsamNEa9gX0,11206
 eizen_nsga/sota/macro_encoding.py,sha256=lwV8Nptwt014UWW-2rAmivYqS-5Jfs3g8DCmzhRzHsA,3037
@@ -8,8 +8,8 @@ eizen_nsga/sota/micro_encoding.py,sha256=LrhgzhgnGMP2Up7Uqy0zT5nedzefqDXl2HJ8YL2
 eizen_nsga/sota/micro_models.py,sha256=piLM6jt4LEiiQuox6AL6I-K0W0XcGLcL2qHWZgPbpvA,22062
 eizen_nsga/sota/micro_operations.py,sha256=3E2JL1eSzWYmtI4X9omBkocWcAoZslXdkIj8CS1h4dQ,8702
 eizen_nsga/sota/model_registry.py,sha256=lQQKUKCVWEQEql5mDZrlwRJEIVaqdokNWEEN4SWrnA0,17750
-eizen_nsga-1.0.1.dist-info/licenses/LICENSE,sha256=_HF_TY-jv6lFXR4QcG1iTnMYgDC3MfslE5S0OvUGlQA,1091
-eizen_nsga-1.0.1.dist-info/METADATA,sha256=D-hn9vHIdz88l9C2MgzxLHZU2nH8gj3DVfSKipCiEdQ,5268
-eizen_nsga-1.0.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-eizen_nsga-1.0.1.dist-info/top_level.txt,sha256=BPCkm-TWk4CpW-H-eKYfaa6KWJuepOHlKR3M5UhB4-4,11
-eizen_nsga-1.0.1.dist-info/RECORD,,
+eizen_nsga-1.0.3.dist-info/LICENSE,sha256=_HF_TY-jv6lFXR4QcG1iTnMYgDC3MfslE5S0OvUGlQA,1091
+eizen_nsga-1.0.3.dist-info/METADATA,sha256=2fcShxz-nwtnc6OU7_PQF2Vx32RPdgqwH4AubYJq8ls,5210
+eizen_nsga-1.0.3.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+eizen_nsga-1.0.3.dist-info/top_level.txt,sha256=BPCkm-TWk4CpW-H-eKYfaa6KWJuepOHlKR3M5UhB4-4,11
+eizen_nsga-1.0.3.dist-info/RECORD,,

{eizen_nsga-1.0.1.dist-info → eizen_nsga-1.0.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.10.2)
+Generator: bdist_wheel (0.42.0)
 Root-Is-Purelib: true
 Tag: py3-none-any