# ViT-CoMer

**Repository Path**: zxnvszyk/ViT-CoMer

## Basic Information

- **Project Name**: ViT-CoMer
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: Co-DETR
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-01-11
- **Last Updated**: 2025-01-11

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

## Applying ViT-CoMer to Co-DETR

### Install
We implement Co-DETR using [MMDetection V2.25.3](https://github.com/open-mmlab/mmdetection/releases/tag/v2.25.3) and [MMCV V1.5.0](https://github.com/open-mmlab/mmcv/releases/tag/v1.5.0).
The source code of MMdetection has been included in this repo and you only need to build MMCV following [official instructions](https://github.com/open-mmlab/mmcv/tree/v1.5.0#installation).
We test our models under ```python=3.7.11,pytorch=1.11.0,cuda=11.3```. Other versions may not be compatible. 

### Data
The COCO dataset and LVIS dataset should be organized as:
```
Co-DETR
└── data
    └── coco
        ├── annotations
        │      ├── instances_train2017.json
        │      └── instances_val2017.json
        ├── train2017
        └── val2017
   
```

### Training
Train ViT-CoMer + Co-DETR with 8 GPUs:
```shell
sh run.sh
```

### Testing
Test ViT-CoMer + Co-DETR with 8 GPUs, and evaluate:
```shell
sh test.sh
```