Metadata-Version: 2.1
Name: argusvision
Version: 1.0.4
Summary: Downloads pretrained Argus Vision models
Home-page: UNKNOWN
License: UNKNOWN
Description: # Argus Vision
        
        argusvision package provides easy to use access to image embedding models pretrained on Bing data - Argus Vision models.
        Interface is based on popular torchvision.
        
        In version 1.0, supported model is Argus Vision V6 - resnext101 32x8d
        
        Please note that provided models are offering image embeddings
        ## Installation
        ``pip install argusvision``
        
        For Windows, torch needs to be install from wheel file.
        Please download the latest torch from here: https://download.pytorch.org/whl/torch_stable.html
        and then install it using: pip install <.whl file>
        
        ## Usage
        Example script:  
        ```
        import argusvision
        import torch
        
        # This will load pretrained model
        model = argusvision.models.resnext101_32x8d()
        
        # This will initialize weights with default values
        model = argusvision.models.resnext101_32x8d(pretrained=False) 
        
        # Load model to CPU memory, interface is the same as torchvision
        model = argusvision.resnext101_32x8d(map_location=torch.device('cpu')) 
        ```
        
        Example of creating image embeddings:
        ```
        import argusvision
        from torchvision import transforms
        import torch
        from PIL import Image
        
        def get_image():
            image = Image.open("example.jpg").convert('RGB')
            preprocess = transforms.Compose([
                transforms.Resize(256),
                transforms.CenterCrop(224),
                transforms.ToTensor(),
                transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
            ])
            return preprocess(image).unsqueeze(0) # Unsqueeze only required when there's 1 image in images batch
        
        model = argusvision.models.resnext101_32x8d(map_location=torch.device('cpu'))
        features = model(get_image())
        print(features.shape)
        ```
        Should output
        ```
        ...
        torch.Size([1, 2048])
        ```
        ### Benchmarks
        Here are the evaluations of the popular datasets  
        Model | CIFAR-10 | STL-10
        --- | --- | ---
        Torchvision, ResNext101 32x8d | 90% | 81.1%
        Argusvision, ResNext101 32x8d | 92.6% | 84.2%
Platform: UNKNOWN
Description-Content-Type: text/markdown
