# pytorch2caffe

**Repository Path**: shixiuyu/pytorch2caffe

## Basic Information

- **Project Name**: pytorch2caffe
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2021-11-26
- **Last Updated**: 2021-11-26

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# pytorch2caffe

A package to convert pytorch model to caffe model. 

## Note

This package supports `F.interpolate` with `mode='bilinear` argument.
It depends on caffe with `Interp` layer.
See [woodsgao/caffe](https://github.com/woodsgao/caffe)

## Installation

`pip3 install pytorch2caffe`

or

    git clone https://github.com/woodsgao/pytorch2caffe
    cd pytorch2caffe
    python3 setup.py install

## Tutorials

Note: import torch first then import pytorch2caffe

    import torch
    from torchvision.models import resnet
    from pytorch2caffe import pytorch2caffe

    name='resnet18'
    resnet18=resnet.resnet18()
    resnet18.eval()
    dummy_input=torch.ones([1,3,224,224])
    pytorch2caffe.trans_net(resnet18, dummy_input, name)
    pytorch2caffe.save_prototxt('{}.prototxt'.format(name))
    pytorch2caffe.save_caffemodel('{}.caffemodel'.format(name))