conda create -n crddc2022 python=3.8
conda activate crddc2022
git clone https://github.com/berry-ding/ShiYu_SeaView_GRDDC2022.git
cd ShiYu_SeaView_GRDDC2022
#cuda >= 11.1
pip3 install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
#official yolov5 r6.1
pip install -r yolov5/requirements.txt
#official yolov7
pip install -r yolov7/requirements.txt
#mmdetection v2.25.0
pip install openmim
mim install mmcv-full==1.6.0
mim install mmcls
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e .
following ipynb
| model | pretrained weight (google drive) |
|---|---|
| YOLOv5x_640 | YOLOv5x_640 |
| YOLOv5x6_1280 | YOLOv5x6_1280 |
| YOLOv7x_640 | YOLOv7x_640 |
| Faster_Swin_l_w7_ms_1and2 | Faster_Swin_l_w7_ms_1and2 |
| Faster_Swin_l_w12_DeformRoI_ms_1and2 | Faster_Swin_l_w12_DeformRoI_ms_1and2 |
| Faster_Swin_l_w12_DeformRoI_ms_3 | Faster_Swin_l_w12_DeformRoI_ms_3 |
| YOLOv5x_1600 | YOLOv5x_1600 |
| YOLOv5m_3200 | YOLOv5m_3200 |