| CUDA | CUDNN | Python | torch | torchvision |
|---|---|---|---|---|
| 11.3 | 8.9.2 | 3.8 | 2.1.2 | 0.16.2 |
| Model | Size | mAPval 0.5:0.95 |
mAPval 0.5 |
Params (M) |
FLOPs (G) |
|---|---|---|---|---|---|
| BYOLO-N | 640 | 38.3 | 55.0 | 1.5 | 3.8 |
| BYOLO-S | 640 | 45.1 | 62.4 | 4.7 | 10.6 |
| BYOLO-M | 640 | 50.0 | 66.7 | 11.1 | 24.7 |
| BYOLO-L | 640 | 52.1 | 69.0 | 20.8 | 46.7 |
Install
git clone https://github.com/Anoue/BYOLO.git
cd BYOLO
pip install -r requirements.txt