https://github.com/jahongir7174/dualdet
DualDet implementation using PyTorch
https://github.com/jahongir7174/dualdet
albumentations coco cutmix mixup mosaic object-detection pytorch testing training
Last synced: 7 months ago
JSON representation
DualDet implementation using PyTorch
- Host: GitHub
- URL: https://github.com/jahongir7174/dualdet
- Owner: jahongir7174
- License: mit
- Created: 2022-09-04T02:39:17.000Z (about 3 years ago)
- Default Branch: master
- Last Pushed: 2023-02-27T00:20:13.000Z (over 2 years ago)
- Last Synced: 2025-01-19T00:51:55.534Z (9 months ago)
- Topics: albumentations, coco, cutmix, mixup, mosaic, object-detection, pytorch, testing, training
- Language: Python
- Homepage: https://arxiv.org/abs/2107.00420
- Size: 16.6 KB
- Stars: 1
- Watchers: 0
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
[DualDet](https://arxiv.org/abs/2107.00420) implementation using PyTorch
### Install
* `pip install mmcv-full`
* `pip install mmdet`
### Train
* `bash ./main.sh ./nets/exp01.py $ --train` for training, `$` is number of GPUs
### Test
* `bash ./main.sh ./nets/exp01.py $ --test` for testing, `$` is number of GPUs
### Results
| Detector | Backbone | Neck | LR Schedule | Box mAP | Config | Download |
|:------------:|:--------:|:----:|:-----------:|--------:|-------------------------:|------------------------------------------------------------------------------------:|
| Faster R-CNN | Swin-T | FPN | 1x | 42.86 | [exp01](./nets/exp01.py) | [model](https://github.com/jahongir7174/DualDet/releases/download/v0.0.1/exp01.pth) |
| DualDet | Swin-T | FPN | 1x | 47.06 | [exp02](./nets/exp02.py) | [model](https://github.com/jahongir7174/DualDet/releases/download/v0.0.1/exp02.pth) |
| Faster R-CNN | Swin-T | FPN | 3x | 45.31 | [exp03](./nets/exp03.py) | [model](https://github.com/jahongir7174/DualDet/releases/download/v0.0.1/exp03.pth) |
| DualDet | Swin-T | FPN | 3x | 49.09 | [exp04](./nets/exp04.py) | [model](https://github.com/jahongir7174/DualDet/releases/download/v0.0.1/exp04.pth) |
### Reference
* https://github.com/ultralytics/yolov5
* https://github.com/open-mmlab/mmdetection