Code Release: CARAFE: Content-Aware ReAssembly of FEatures (ICCV 2019) (#1583)
* add carafe ops * rename carafe benchmark * grad check fix * update grad check * update grad check output * add fpn carafe & mask head carafe * add ReadMe * update readme * add carafe setup * update naive carafe * update readme and setup * readme typo fix * fix flake8 error * fix flake 8 error * fix flake 8 * fix flake 8 more * flake 8 fix plus * flake 8 fix * fix flake 8 * reformat ops files * update fpn files and cfgs * update readme * update fcn_mask_head * update fpn_carafe * update kernel * update * update * add docstring in FPN_CARAFE * reformat with yapf * update * update * add build upsampler * fix mask head build error * reformat build upsample layer * add doc string for CARAFE and PixelShuffle * update * update upsample_cfg_ * update * update doc string * rm abbr in build upsample layer * update readme * update model_zoo * add link to other features in ReadMe
Showing
- README.md 7 additions, 6 deletionsREADME.md
- configs/carafe/README.md 53 additions, 0 deletionsconfigs/carafe/README.md
- configs/carafe/faster_rcnn_r50_fpn_carafe_1x.py 188 additions, 0 deletionsconfigs/carafe/faster_rcnn_r50_fpn_carafe_1x.py
- configs/carafe/mask_rcnn_r50_fpn_carafe_1x.py 210 additions, 0 deletionsconfigs/carafe/mask_rcnn_r50_fpn_carafe_1x.py
- docs/MODEL_ZOO.md 3 additions, 0 deletionsdocs/MODEL_ZOO.md
- mmdet/models/mask_heads/fcn_mask_head.py 34 additions, 17 deletionsmmdet/models/mask_heads/fcn_mask_head.py
- mmdet/models/necks/__init__.py 2 additions, 1 deletionmmdet/models/necks/__init__.py
- mmdet/models/necks/fpn_carafe.py 254 additions, 0 deletionsmmdet/models/necks/fpn_carafe.py
- mmdet/models/utils/__init__.py 3 additions, 2 deletionsmmdet/models/utils/__init__.py
- mmdet/models/utils/upsample.py 78 additions, 0 deletionsmmdet/models/utils/upsample.py
- mmdet/ops/carafe/__init__.py 3 additions, 0 deletionsmmdet/ops/carafe/__init__.py
- mmdet/ops/carafe/carafe.py 237 additions, 0 deletionsmmdet/ops/carafe/carafe.py
- mmdet/ops/carafe/grad_check.py 61 additions, 0 deletionsmmdet/ops/carafe/grad_check.py
- mmdet/ops/carafe/setup.py 29 additions, 0 deletionsmmdet/ops/carafe/setup.py
- mmdet/ops/carafe/src/carafe_cuda.cpp 113 additions, 0 deletionsmmdet/ops/carafe/src/carafe_cuda.cpp
- mmdet/ops/carafe/src/carafe_cuda_kernel.cu 475 additions, 0 deletionsmmdet/ops/carafe/src/carafe_cuda_kernel.cu
- mmdet/ops/carafe/src/carafe_naive_cuda.cpp 75 additions, 0 deletionsmmdet/ops/carafe/src/carafe_naive_cuda.cpp
- mmdet/ops/carafe/src/carafe_naive_cuda_kernel.cu 176 additions, 0 deletionsmmdet/ops/carafe/src/carafe_naive_cuda_kernel.cu
- setup.py 11 additions, 0 deletionssetup.py
Loading
Please register or sign in to comment