imagenet-classes 0.1.1__tar.gz

imagenet_classes-0.1.1/.gitignore

@@ -0,0 +1,180 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fm/#using-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be added to the global gitignore or merged into this project gitignore.  For a PyCharm
+#  project, it is recommended to include the following files:
+#  .idea/
+#  *.iml
+#  *.ipr
+#  *.iws
+
+# VS Code
+.vscode/
+
+# macOS
+.DS_Store
+
+# Windows
+Thumbs.db
+ehthumbs.db
+Desktop.ini
+
+# Linux
+*~
+.fuse_hidden*
+.directory
+.Trash-*
+.nfs*

imagenet_classes-0.1.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Illia Volkov, Nikita Kisel
+
+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.

imagenet_classes-0.1.1/MANIFEST.in

@@ -0,0 +1,19 @@
+include README.md
+include LICENSE
+include requirements.txt
+include pyproject.toml
+include setup.py
+
+recursive-include class_mapping *.npy
+recursive-include class_mapping *.py
+
+recursive-exclude * __pycache__
+recursive-exclude * *.py[co]
+recursive-exclude * .DS_Store
+recursive-exclude * .git*
+recursive-exclude * .pytest_cache
+recursive-exclude * .coverage
+recursive-exclude * htmlcov
+recursive-exclude * dist
+recursive-exclude * build
+recursive-exclude * *.egg-info

imagenet_classes-0.1.1/PKG-INFO

@@ -0,0 +1,90 @@
+Metadata-Version: 2.4
+Name: imagenet-classes
+Version: 0.1.1
+Summary: A Python package for managing and retrieving ImageNet-1k mappings among integer class IDs, string class IDs, and human-readable class names.
+Home-page: https://github.com/gonikisgo/imagenet-classes
+Author: Illia Volkov, Nikita Kisel
+Author-email: Illia Volkov <volkoill@cvut.cz>, Nikita Kisel <kiselnik@fel.cvut.cz>
+Maintainer-email: Nikita Kisel <kiselnik@fel.cvut.cz>
+License: MIT
+Project-URL: Homepage, https://github.com/gonikisgo/imagenet-classes
+Project-URL: Documentation, https://github.com/gonikisgo/imagenet-classes#readme
+Project-URL: Repository, https://github.com/gonikisgo/imagenet-classes
+Project-URL: Bug Tracker, https://github.com/gonikisgo/imagenet-classes/issues
+Keywords: imagenet,computer-vision,machine-learning,deep-learning,classification
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Typing :: Typed
+Classifier: Framework :: Pytest
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy>=1.19.0
+Provides-Extra: dev
+Requires-Dist: pytest>=6.0; extra == "dev"
+Requires-Dist: pytest-cov>=2.0; extra == "dev"
+Requires-Dist: black>=21.0; extra == "dev"
+Requires-Dist: flake8>=3.8; extra == "dev"
+Requires-Dist: mypy>=0.800; extra == "dev"
+Dynamic: author
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
+
+# ImageNet Classes
+
+A Python package for managing and retrieving ImageNet-1k (ImageNet2012) class names and mappings.
+
+## Installation
+
+### From PyPI (when published)
+```bash
+pip install imagenet-classes
+```
+
+### From Source
+```bash
+git clone https://github.com/gonikisgo/imagenet-classes.git
+cd imagenet-classes
+pip install -e .
+```
+
+## Quick Start
+
+```python
+from class_mapping import ClassDictionary
+
+# Initialize the class dictionary
+class_dict = ClassDictionary()
+
+# Get class name for ImageNet-1k class index 0
+class_name = class_dict.get_class_name(0)
+print(f"Class 0: {class_name}")
+
+# Get ImageNet21k to ImageNet-1k class mapping
+class_1k = class_dict.get_class_1k("n01440764")
+print(f"Class 1k mapping: {class_1k}")
+
+# Get custom class name
+custom_name = class_dict.get_custom_class_name(0)
+print(f"Custom name: {custom_name}")
+```
+
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
+
+## Support
+
+For support, please open an issue on the [GitHub repository](https://github.com/gonikisgo/imagenet-classes/issues).

imagenet_classes-0.1.1/README.md

@@ -0,0 +1,46 @@
+# ImageNet Classes
+
+A Python package for managing and retrieving ImageNet-1k (ImageNet2012) class names and mappings.
+
+## Installation
+
+### From PyPI (when published)
+```bash
+pip install imagenet-classes
+```
+
+### From Source
+```bash
+git clone https://github.com/gonikisgo/imagenet-classes.git
+cd imagenet-classes
+pip install -e .
+```
+
+## Quick Start
+
+```python
+from class_mapping import ClassDictionary
+
+# Initialize the class dictionary
+class_dict = ClassDictionary()
+
+# Get class name for ImageNet-1k class index 0
+class_name = class_dict.get_class_name(0)
+print(f"Class 0: {class_name}")
+
+# Get ImageNet21k to ImageNet-1k class mapping
+class_1k = class_dict.get_class_1k("n01440764")
+print(f"Class 1k mapping: {class_1k}")
+
+# Get custom class name
+custom_name = class_dict.get_custom_class_name(0)
+print(f"Custom name: {custom_name}")
+```
+
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
+
+## Support
+
+For support, please open an issue on the [GitHub repository](https://github.com/gonikisgo/imagenet-classes/issues).

imagenet_classes-0.1.1/class_mapping/__init__.py

@@ -0,0 +1,13 @@
+"""
+ImageNet Classes Package
+
+A Python package for managing and retrieving ImageNet-1k mappings among integer class IDs, string class IDs, and human-readable class names.
+"""
+
+from .class_loader import ClassDictionary
+
+__version__ = "0.1.1"
+__author__ = "Illia Volkov, Nikita Kisel"
+__email__ = "kiselnik@fel.cvut.cz"
+
+__all__ = ["ClassDictionary"]

imagenet_classes-0.1.1/class_mapping/class_loader.py

@@ -0,0 +1,187 @@
+import os
+
+import numpy as np
+
+
+class ClassDictionary:
+    """
+    Class to manage and retrieve class names from files containing ImageNet class data.
+
+    Attributes:
+        filename (str): The path to the file containing class-name mapping for ImageNet2012.
+        filename_21k (str): The path to the file containing the class mapping from ImageNet21k to ImageNet2012.
+        class2name (dict): A dictionary to store the class data for ImageNet2012.
+        class2short_class (dict): A dictionary to store the class data for ImageNet21k.
+    """
+    def __init__(self, filename = 'imagenet2012_classes.npy', custom_cls_filename='cls_name_mod.npy',
+                 filename_21k = 'imagenet21k_classes.npy', filename_21k_r = 'imagenet21k_classes_r.npy',
+                 filename_val_class = 'val_img_classes_pairs.npy'):
+        current_dir = os.path.dirname(__file__)
+        self.filename = os.path.join(current_dir, filename)
+        self.custom_cls_filename = os.path.join(current_dir, custom_cls_filename)
+        self.filename_21k = os.path.join(current_dir, filename_21k)
+        self.filename_21k_r = os.path.join(current_dir, filename_21k_r)
+        self.filename_val_class = os.path.join(current_dir, filename_val_class)
+        self.class2name = {}
+        self.class2custom_name = None
+        self.class2short_class = {}
+        self.class2short_class_r = {}
+        self.val2short_class = {}
+        self._data_loaded = False
+        self._custom_data_loaded = False
+        self._data_21k_loaded = False
+        self._data_21k_r_loaded = False
+        self._val_class_loaded = False
+
+    def __load_class2name_mapping(self):
+        """Loads class data from the file into the class2name dictionary."""
+        try:
+            self.class2name = np.load(self.filename, allow_pickle=True).item()
+            self._data_loaded = True
+        except FileNotFoundError:
+            print(f"Error: The file '{self.filename}' was not found.")
+            self._data_loaded = False
+        except ValueError:
+            print(f"Error: The file '{self.filename}' contains invalid data.")
+            self._data_loaded = False
+        except Exception as e:
+            print(f"An unexpected error occurred: {e}")
+            self._data_loaded = False
+
+    def __load_class2custom_name_mapping(self):
+        """Loads class data from the file into the class2custom_name dictionary."""
+        try:
+            self.class2custom_name = np.load(self.custom_cls_filename)
+            self._custom_data_loaded = True
+        except FileNotFoundError:
+            print(f"Error: The file '{self.custom_cls_filename}' was not found.")
+            self._data_loaded = False
+        except ValueError:
+            print(f"Error: The file '{self.custom_cls_filename}' contains invalid data.")
+            self._data_loaded = False
+        except Exception as e:
+            print(f"An unexpected error occurred: {e}")
+            self._custom_data_loaded = False
+
+    def __load_class2short_class_mapping(self):
+        """Loads class data from the file into the class2short_class dictionary."""
+        try:
+            self.class2short_class = np.load(self.filename_21k, allow_pickle=True).item()
+            self._data_21k_loaded = True
+        except FileNotFoundError:
+            print(f"Error: The file '{self.filename_21k}' was not found.")
+            self._data_21k_loaded = False
+        except ValueError:
+            print(f"Error: The file '{self.filename_21k}' contains invalid data.")
+            self._data_21k_loaded = False
+        except Exception as e:
+            print(f"An unexpected error occurred: {e}")
+            self._data_21k_loaded = False
+
+    def __load_class2short_class_r_mapping(self):
+        """Loads numeric class data from the file into the class2short_class dictionary."""
+        try:
+            self.class2short_class_r = np.load(self.filename_21k_r, allow_pickle=True).item()
+            self._data_21k_r_loaded = True
+        except FileNotFoundError:
+            print(f"Error: The file '{self.filename_21k_r}' was not found.")
+            self._data_21k_r_loaded = False
+        except ValueError:
+            print(f"Error: The file '{self.filename_21k_r}' contains invalid data.")
+            self._data_21k_r_loaded = False
+        except Exception as e:
+            print(f"An unexpected error occurred: {e}")
+            self._data_21k_r_loaded = False
+
+    def __load_val2short_class_mapping(self):
+        """Loads validation images data from the file into the val2short_class dictionary."""
+        try:
+            self.val2short_class = np.load(self.filename_val_class, allow_pickle=True).item()
+            self._val_class_loaded = True
+        except FileNotFoundError:
+            print(f"Error: The file '{self.filename_val_class}' was not found.")
+            self._val_class_loaded = False
+        except ValueError:
+            print(f"Error: The file '{self.filename_val_class}' contains invalid data.")
+            self._val_class_loaded = False
+        except Exception as e:
+            print(f"An unexpected error occurred: {e}")
+            self._val_class_loaded = False
+
+    def get_class_name(self, key):
+        """
+        Retrieves the class names associated with a given key.
+
+        Args:
+            key (int): The key for which to retrieve class names.
+
+        Returns:
+            list[str] | None: A list of class names if the key exists, None otherwise.
+        """
+        if not self._data_loaded:
+            self.__load_class2name_mapping()
+        return self.class2name.get(key)
+
+    def get_custom_class_name(self, key):
+        """
+        Retrieves the custom class names associated with a given key.
+
+        Args:
+            key (int): The key for which to retrieve class names.
+
+        Returns:
+            str | None: A class name if the key exists, None otherwise.
+        """
+        if not self._custom_data_loaded:
+            self.__load_class2custom_name_mapping()
+        if self.class2custom_name is None:
+            return None
+        return self.class2custom_name[key]
+
+    def get_class_1k(self, key):
+        """
+        Retrieves the class names associated with a given key from the ImageNet21k file.
+
+        Args:
+            key (str): The key for which to perform mapping.
+
+        Returns:
+            int | None: class label in imagenet2012 if the key exists, None otherwise.
+        """
+        if not self._data_21k_loaded:
+            self.__load_class2short_class_mapping()
+        return self.class2short_class.get(key)
+
+    def get_class_1k_r(self, key):
+        """
+        Retrieves the numeric class names associated with a given key from the ImageNet21k file.
+
+        Args:
+            key (int): The key for which to perform mapping.
+
+        Returns:
+            str | None: class label in imagenet21k if the key exists, None otherwise.
+        """
+        if not self._data_21k_r_loaded:
+            self.__load_class2short_class_r_mapping()
+        return self.class2short_class_r.get(key)
+
+    def get_val_img_class(self, key):
+        """
+        Retrieves the numeric class names associated with a given image key from the ImageNet21k file.
+
+        Args:
+            key (str): The validation split image id.
+
+        Returns:
+            str | None: class label in imagenet21k if the key exists, None otherwise.
+        """
+        if not self._val_class_loaded:
+            self.__load_val2short_class_mapping()
+        return self.val2short_class.get(key)
+
+    def create_cls_name_dict(self, class_names):
+        return {idx: cls for idx, cls in enumerate(class_names)}
+
+    def create_im_to_orig(self, class_list):
+        return {cls: idx for idx, cls in enumerate(class_list)}

imagenet_classes-0.1.1/examples/basic_usage.py

@@ -0,0 +1,59 @@
+#!/usr/bin/env python3
+"""
+Basic usage example for the imagenet-classes package.
+"""
+
+from class_mapping import ClassDictionary
+
+
+def main():
+    """Demonstrate basic usage of the ClassDictionary class."""
+    
+    print("ImageNet Classes Package - Basic Usage Example")
+    print("=" * 50)
+    
+    # Initialize the class dictionary
+    print("Initializing ClassDictionary...")
+    class_dict = ClassDictionary()
+    
+    # Demonstrate utility methods
+    print("\n1. Creating class name dictionaries:")
+    sample_classes = ['tench', 'goldfish', 'great_white_shark', 'tiger_shark', 'hammerhead']
+    
+    # Create index to class name mapping
+    idx_to_name = class_dict.create_cls_name_dict(sample_classes)
+    print(f"   Index to name: {idx_to_name}")
+    
+    # Create class name to index mapping
+    name_to_idx = class_dict.create_im_to_orig(sample_classes)
+    print(f"   Name to index: {name_to_idx}")
+    
+    print("\n2. Accessing class data:")
+    print("   Note: Data files (.npy) need to be present for these methods to work")
+    
+    # Try to get class names (will return None if files don't exist)
+    class_name = class_dict.get_class_name(0)
+    print(f"   Class 0: {class_name}")
+    
+    # Try to get custom class names
+    custom_name = class_dict.get_custom_class_name(0)
+    print(f"   Custom class 0: {custom_name}")
+    
+    # Try to get ImageNet21k mappings
+    class_1k = class_dict.get_class_1k("n01440764")
+    print(f"   ImageNet21k mapping for n01440764: {class_1k}")
+    
+    print("\n3. File paths:")
+    print(f"   ImageNet2012 classes: {class_dict.filename}")
+    print(f"   ImageNet21k classes: {class_dict.filename_21k}")
+    print(f"   Custom classes: {class_dict.custom_cls_filename}")
+    
+    print("\nExample completed successfully!")
+    print("\nTo use with actual data:")
+    print("1. Ensure the required .npy files are in the class_mapping directory")
+    print("2. The package will automatically load data when methods are called")
+    print("3. Data is loaded lazily for optimal memory usage")
+
+
+if __name__ == "__main__":
+    main()

imagenet_classes-0.1.1/imagenet_classes.egg-info/PKG-INFO

@@ -0,0 +1,90 @@
+Metadata-Version: 2.4
+Name: imagenet-classes
+Version: 0.1.1
+Summary: A Python package for managing and retrieving ImageNet-1k mappings among integer class IDs, string class IDs, and human-readable class names.
+Home-page: https://github.com/gonikisgo/imagenet-classes
+Author: Illia Volkov, Nikita Kisel
+Author-email: Illia Volkov <volkoill@cvut.cz>, Nikita Kisel <kiselnik@fel.cvut.cz>
+Maintainer-email: Nikita Kisel <kiselnik@fel.cvut.cz>
+License: MIT
+Project-URL: Homepage, https://github.com/gonikisgo/imagenet-classes
+Project-URL: Documentation, https://github.com/gonikisgo/imagenet-classes#readme
+Project-URL: Repository, https://github.com/gonikisgo/imagenet-classes
+Project-URL: Bug Tracker, https://github.com/gonikisgo/imagenet-classes/issues
+Keywords: imagenet,computer-vision,machine-learning,deep-learning,classification
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Typing :: Typed
+Classifier: Framework :: Pytest
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy>=1.19.0
+Provides-Extra: dev
+Requires-Dist: pytest>=6.0; extra == "dev"
+Requires-Dist: pytest-cov>=2.0; extra == "dev"
+Requires-Dist: black>=21.0; extra == "dev"
+Requires-Dist: flake8>=3.8; extra == "dev"
+Requires-Dist: mypy>=0.800; extra == "dev"
+Dynamic: author
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
+
+# ImageNet Classes
+
+A Python package for managing and retrieving ImageNet-1k (ImageNet2012) class names and mappings.
+
+## Installation
+
+### From PyPI (when published)
+```bash
+pip install imagenet-classes
+```
+
+### From Source
+```bash
+git clone https://github.com/gonikisgo/imagenet-classes.git
+cd imagenet-classes
+pip install -e .
+```
+
+## Quick Start
+
+```python
+from class_mapping import ClassDictionary
+
+# Initialize the class dictionary
+class_dict = ClassDictionary()
+
+# Get class name for ImageNet-1k class index 0
+class_name = class_dict.get_class_name(0)
+print(f"Class 0: {class_name}")
+
+# Get ImageNet21k to ImageNet-1k class mapping
+class_1k = class_dict.get_class_1k("n01440764")
+print(f"Class 1k mapping: {class_1k}")
+
+# Get custom class name
+custom_name = class_dict.get_custom_class_name(0)
+print(f"Custom name: {custom_name}")
+```
+
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
+
+## Support
+
+For support, please open an issue on the [GitHub repository](https://github.com/gonikisgo/imagenet-classes/issues).

imagenet_classes-0.1.1/imagenet_classes.egg-info/SOURCES.txt

@@ -0,0 +1,21 @@
+.gitignore
+LICENSE
+MANIFEST.in
+README.md
+pyproject.toml
+requirements.txt
+setup.py
+class_mapping/__init__.py
+class_mapping/class_loader.py
+class_mapping/cls_name_mod.npy
+class_mapping/imagenet2012_classes.npy
+class_mapping/imagenet21k_classes.npy
+class_mapping/imagenet21k_classes_r.npy
+class_mapping/val_img_classes_pairs.npy
+examples/basic_usage.py
+imagenet_classes.egg-info/PKG-INFO
+imagenet_classes.egg-info/SOURCES.txt
+imagenet_classes.egg-info/dependency_links.txt
+imagenet_classes.egg-info/not-zip-safe
+imagenet_classes.egg-info/requires.txt
+imagenet_classes.egg-info/top_level.txt

imagenet_classes-0.1.1/imagenet_classes.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

imagenet_classes-0.1.1/imagenet_classes.egg-info/not-zip-safe

@@ -0,0 +1 @@
+

imagenet_classes-0.1.1/imagenet_classes.egg-info/requires.txt

@@ -0,0 +1,8 @@
+numpy>=1.19.0
+
+[dev]
+pytest>=6.0
+pytest-cov>=2.0
+black>=21.0
+flake8>=3.8
+mypy>=0.800

imagenet_classes-0.1.1/imagenet_classes.egg-info/top_level.txt

@@ -0,0 +1 @@
+class_mapping

imagenet_classes-0.1.1/pyproject.toml

@@ -0,0 +1,121 @@
+[build-system]
+requires = ["setuptools>=45", "wheel", "setuptools_scm[toml]>=6.2"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "imagenet-classes"
+dynamic = ["version"]
+description = "A Python package for managing and retrieving ImageNet-1k mappings among integer class IDs, string class IDs, and human-readable class names."
+readme = "README.md"
+license = {text = "MIT"}
+authors = [
+    {name = "Illia Volkov", email = "volkoill@cvut.cz"},
+    {name = "Nikita Kisel", email = "kiselnik@fel.cvut.cz"}
+]
+maintainers = [
+    {name = "Nikita Kisel", email = "kiselnik@fel.cvut.cz"}
+]
+keywords = ["imagenet", "computer-vision", "machine-learning", "deep-learning", "classification"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Science/Research",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Typing :: Typed",
+    "Framework :: Pytest",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+requires-python = ">=3.7"
+dependencies = [
+    "numpy>=1.19.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=6.0",
+    "pytest-cov>=2.0",
+    "black>=21.0",
+    "flake8>=3.8",
+    "mypy>=0.800",
+]
+
+[project.urls]
+Homepage = "https://github.com/gonikisgo/imagenet-classes"
+Documentation = "https://github.com/gonikisgo/imagenet-classes#readme"
+Repository = "https://github.com/gonikisgo/imagenet-classes"
+"Bug Tracker" = "https://github.com/gonikisgo/imagenet-classes/issues"
+
+[tool.setuptools]
+packages = ["class_mapping"]
+include-package-data = true
+
+[tool.setuptools.package-data]
+class_mapping = ["*.npy"]
+
+[tool.setuptools.dynamic]
+version = {attr = "class_mapping.__version__"}
+
+[tool.black]
+line-length = 88
+target-version = ['py37', 'py38', 'py39', 'py310', 'py311']
+include = '\.pyi?$'
+extend-exclude = '''
+/(
+  # directories
+  \.eggs
+  | \.git
+  | \.hg
+  | \.mypy_cache
+  | \.tox
+  | \.venv
+  | build
+  | dist
+)/
+'''
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+python_classes = ["Test*"]
+python_functions = ["test_*"]
+addopts = [
+    "--strict-markers",
+    "--strict-config",
+    "--cov=class_mapping",
+    "--cov-report=term-missing",
+    "--cov-report=html",
+]
+markers = [
+    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
+    "integration: marks tests as integration tests",
+]
+
+[tool.coverage.run]
+source = ["class_mapping"]
+omit = [
+    "*/tests/*",
+    "*/test_*",
+    "*/__pycache__/*",
+]
+
+[tool.coverage.report]
+exclude_lines = [
+    "pragma: no cover",
+    "def __repr__",
+    "if self.debug:",
+    "if settings.DEBUG",
+    "raise AssertionError",
+    "raise NotImplementedError",
+    "if 0:",
+    "if __name__ == .__main__.:",
+    "class .*\\bProtocol\\):",
+    "@(abc\\.)?abstractmethod",
+]

imagenet_classes-0.1.1/requirements.txt

@@ -0,0 +1 @@
+numpy>=1.19.0

imagenet_classes-0.1.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

imagenet_classes-0.1.1/setup.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+"""
+Setup script for the imagenet-classes package.
+"""
+
+from setuptools import setup, find_packages
+import os
+
+# Read the README file for long description
+def read_readme():
+    with open("README.md", "r", encoding="utf-8") as fh:
+        return fh.read()
+
+# Read requirements from requirements.txt if it exists
+def read_requirements():
+    requirements = []
+    if os.path.exists("requirements.txt"):
+        with open("requirements.txt", "r", encoding="utf-8") as fh:
+            requirements = [line.strip() for line in fh if line.strip() and not line.startswith("#")]
+    return requirements
+
+setup(
+    name="imagenet-classes",
+    author="Illia Volkov, Nikita Kisel",
+    author_email="kiselnik@fel.cvut.cz",  # Primary contact
+    description="A Python package for managing and retrieving ImageNet-1k mappings among integer class IDs, string class IDs, and human-readable class names.",
+    long_description=read_readme(),
+    long_description_content_type="text/markdown",
+    url="https://github.com/gonikisgo/imagenet-classes",
+    packages=find_packages(),
+    classifiers=[
+        "Development Status :: 4 - Beta",
+        "Intended Audience :: Developers",
+        "Intended Audience :: Science/Research",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+        "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.7",
+        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
+        "Topic :: Scientific/Engineering :: Artificial Intelligence",
+        "Topic :: Software Development :: Libraries :: Python Modules",
+    ],
+    python_requires=">=3.8",
+    install_requires=[
+        "numpy>=1.19.0",
+    ],
+    extras_require={
+        "dev": [
+            "pytest>=6.0",
+            "pytest-cov>=2.0",
+            "black>=21.0",
+            "flake8>=3.8",
+            "mypy>=0.800",
+        ],
+    },
+    include_package_data=True,
+    package_data={
+        "class_mapping": [
+            "*.npy",
+        ],
+    },
+    zip_safe=False,
+    keywords="imagenet, computer-vision, machine-learning, deep-learning, classification",
+    project_urls={
+        "Bug Reports": "https://github.com/gonikisgo/imagenet-classes/issues",
+        "Source": "https://github.com/gonikisgo/imagenet-classes",
+        "Documentation": "https://github.com/gonikisgo/imagenet-classes#readme",
+    },
+)