Metadata-Version: 1.1
Name: image_classification_pytorch
Version: 0.0.2
Summary: Image classification with pretrained models in Pytorch
Home-page: https://github.com/denred0/image_classification_pytorch
Author: Denis Potapov
Author-email: potapovdenisdmit@gmail.com
License: MIT
Download-URL: https://github.com/denred0/image_classification_pytorch/archive/refs/tags/0.0.1.tar.gz
Description: # Image classification with pretrained models in Pytorch
        
        Use pretrained models to train your data.
        
        ## Installation
        
        ```sh
        pip install image-classification-pytorch
        ```
        
        ## Example
        
        ```sh
        tf_efficientnet_b4_ns = {'model_type': 'tf_efficientnet_b4_ns', 'im_size': 380, 'im_size_test': 380, 'batch_size': 8, 'mean': [0.485, 0.456, 0.406], 'std': [0.229, 0.224, 0.225]}
        models = [tf_efficientnet_b4_ns]
        trainer = ICPTrainer(models=models, data_dir='data_simpsons')
        trainer.fit_test()
        ```
Keywords: pytorch,image classification,imagenet,pretrained model
Platform: UNKNOWN
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Topic :: Software Development :: Build Tools
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3.6
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
