Skip to content

Latest commit

 

History

History
executable file
·
33 lines (27 loc) · 852 Bytes

README.md

File metadata and controls

executable file
·
33 lines (27 loc) · 852 Bytes
Raw

CARAFE

A PyTorch implementation of CARAFE based on ICCV 2019 paper CARAFE: Content-Aware ReAssembly of FEatures.

Network Architecture image from the paper

Requirements

conda install pytorch torchvision -c pytorch

Installation

pip install git+https://github.com/leftthomas/CARAFE.git@master

To update:

pip install --upgrade git+https://github.com/leftthomas/CARAFE.git@master

Example

from carafe import CARAFE, CARAFENaive
module = CARAFE(kernel_size=5, group_size=1, scale_factor=4)
module = CARAFENaive(kernel_size=5, group_size=1, scale_factor=4)

Contribution

Any contributions to CARAFE are welcome!

Copyright and License

CARAFE is provided under the MIT License.