Skip to content

phv2312/rtmdet_ship_detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Ship Detection

Installation

conda create --name mmd python=3.8
conda activate mmd
conda install "pytorch==1.10.1" "torchvision==0.11.2" "cudatoolkit==10.2.*" -c pytorch -y

pip install openmim
mim install "mmengine==0.8.4"
mim install "mmcv==2.0.1"
mim install "mmdet==3.0.0"
 
git clone https://github.com/open-mmlab/mmrotate.git
cd mmrotate
git checkout "d50ab76"
pip install -v -e .

Or

conda create --name rtmdet python=3.8 -y
conda activate rtmdet
conda install pytorch==1.12.1 torchvision cudatoolkit -c pytorch
pip install openmim==0.3.9
mim install mmcv-full==1.7.2
mim install mmdet==2.28.2
pip install "mmrotate @ git+ssh://[email protected]/open-mmlab/[email protected]"

Sample usage

Training

conda activate mmd
python train.py

Inference

conda activate mmd
python infer.py

About

Ship detection using rtmdet

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages