Metadata-Version: 2.1
Name: om-simple
Version: 0.1.6
Summary: OM Simple
Home-page: https://www.soco.ai
Author: kyusonglee
License: UNKNOWN
Description: # Image Classification
        ```
        pip install om-simple
        ```
        
        ## Data Preparation
        
        ```
        {dataset name}/
        ├── train/
        │   ├── {class1}/
        │   ├── {class2}/
        │   ├── ...
        └── val/
            ├── {class1}/
            ├── {class2}/
            ├── ...
        ```
        
        ## Example Code
        ```python
        from om_simple.img_class_head import ImageClassification
        from om_simple.img_class_model import run_train
        from om_simple.tools.utils import is_blur
        
        # Train
        run_train("{dataset name}",
            model_name="{model_name}", # e.g., resnet18, resnet50, vit_base_patch16_224, etc....
            outdir="{output_dir}")
        
        # Simple Classification
        X = ImageClassification("epoch099.ckpt")
        z = X.predict(images=["sample.jpg"])
        
        
        # Blur detection
        print (is_blur("sample.jpg"))
        
        # Multi label classification
        
        from om_simple.multi_class_model import MultiClass
        X = MultiClass("model.ckpt","label.json")
        z = X.predict(images=["sample.jpg"])
        
        
        
        
        ```
        
        
        # How to get available model_name
        ```
        import timm
        avail_pretrained_models = timm.list_models(pretrained=True)
        print (avail_pretrained_models)
        
        all_vit_models = timm.list_models('vit*')
        print (all_vit_models)
        
        ```
        
        # Tensorboard
        ```
        tensorboard --logdir {output_dir} --bind_all
        ```
        
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: Free for non-commercial use
Classifier: Operating System :: OS Independent
Description-Content-Type: text/markdown
