Metadata-Version: 2.1
Name: CVX2
Version: 0.0.2
Summary: Tools of CV(Computer Vision)
Home-page: https://gitee.com/summry/cvx2
Author: summy
Author-email: xiazhongbiao@126.com
License: UNKNOWN
Keywords: CV,Computer Vision,Machine learning,Deep learning,torch
Platform: UNKNOWN
Requires-Python: >=3.6
Description-Content-Type: text/markdown
License-File: LICENSE

Usage Sample
''''''''''''

.. code:: python

        import torch
        from torch import nn
        from cvx2 import WidthBlock

        model = nn.Sequential(
            WidthBlock(c1=1, c2=32),
            nn.MaxPool2d(kernel_size=2, stride=2),
            WidthBlock(c1=32, c2=64),
            nn.MaxPool2d(kernel_size=2, stride=2),
            nn.Flatten(),
            nn.Linear(in_features=64*49, out_features=1024),
            nn.Dropout(0.2),
            nn.SiLU(inplace=True),
            nn.Linear(in_features=1024, out_features=2),
        )

        img = torch.randn(1, 1, 28, 28)
        print(model(img).shape)


