# Code for HiViT

### 1. Self-supervised pre-training and ImageNet-1K fine-tuning

please refer to [hivit_ssl/README.md](hivit_ssl/README.md)

### 2. COCO detection

We provide a mmdetection based codebase in [hivit_mmdet/](hivit_mmdet/) and a [configuration file](hivit_mmdet/configs/_hivit_/hivit_base_win14_mask_rcnn_fpn_3x_coco_ld_auto.py)

### 3. ADE20K semantic segmentation

Our segmentation code based on [BEiT](https://github.com/microsoft/unilm/tree/master/beit), The configuration file for base model at [here](hivit_mmseg/configs/upernet_hivit_base_640_80k_ade20k_bs32_geo_ms.py). 
