Data

Tools

Indexes

Applications

Experiments

Open Source Science

avm

https://github.com/yuxueyang1204/avm

Science Score: 26.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • ○
    CITATION.cff file
  • ✓
    codemeta.json file
    Found codemeta.json file
  • ✓
    .zenodo.json file
    Found .zenodo.json file
  • ○
    DOI references
  • ○
    Academic publication links
  • ○
    Academic email domains
  • ○
    Institutional organization owner
  • ○
    JOSS paper metadata
  • ○
    Scientific vocabulary similarity
    Low similarity (3.7%) to scientific vocabulary
Last synced: 10 months ago · JSON representation

Repository

Basic Info
  • Host: GitHub
  • Owner: YuxueYang1204
  • License: apache-2.0
  • Language: Python
  • Default Branch: main
  • Size: 29.5 MB
Statistics
  • Stars: 0
  • Watchers: 1
  • Forks: 0
  • Open Issues: 0
  • Releases: 0
Created about 3 years ago · Last pushed about 3 years ago
Metadata Files
Readme License Citation

README.md

AVM

环境配置

  1. 使用 MIM 安装 MMCV 和 MMDetection

shell pip install -U openmim mim install mmcv-full mim install mmdet\<3.0.0

  1. 安装 MMSegmentation

shell pip install mmsegmentation

  1. 安装 MMRotate

shell git clone https://github.com/Yuppie898988/AVM.git cd AVM pip install -v -e .

数据准备

AVM原始数据集目录需调整为如下结构

python AVM └─ data └─ avm ├─ images │ ├─ b2_left │ │ └─ avm │ │ └─ *.jpg │ ├─ b2_right │ ├─ b2_to_b3 │ └─ b3_to_b2 ├─ det_annotations │ ├─ b2_left │ │ └─ *.json │ ├─ b2_right │ ├─ b2_to_b3 │ └─ b3_to_b2 └─ mask ├─ b2_left │ └─ *.png ├─ b2_right ├─ b2_to_b3 └─ b3_to_b2

通过create_data.py将数据集转换为指定格式

shell python tools/create_data.py --avm-dir ./data/avm --output-dir ./data/avm_seg_det

转换后格式为

python AVM └─ data ├─ avm │ └─ ... └─ avm_seg_det └─ train ├─ annfiles │ └─ *.txt ├─ images │ └─ *.png └─ segments └─ *.png

训练

shell ./tools/dist_train.sh configs/avm/seg_rotated_retinanet_obb_r50_fpn_1x_avm_le90.py 8 --work-dir ${YOUR_WORK_DIR}

推理

shell python tools/inference.py --config configs/avm/seg_rotated_retinanet_obb_r50_fpn_1x_avm_le90.py --checkpoint ${CHECKPOINT_FILE} --image_path ${IMAGE_FILE} --save_dir ${SAVE_DIR_PATH}

样例

|Input|Detection|Segmentation| |-|-|-| ||||

Owner

  • Name: Yuxue Yang
  • Login: YuxueYang1204
  • Kind: user
  • Location: Beijing
  • Company: CASIA

GitHub Events

Total
Last Year

Dependencies

docker/Dockerfile docker
  • pytorch/pytorch ${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel build
docker/serve/Dockerfile docker
  • pytorch/pytorch ${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel build
requirements/build.txt pypi
  • cython *
  • numpy *
requirements/docs.txt pypi
  • docutils ==0.16.0
  • markdown >=3.4.0
  • myst-parser *
  • sphinx ==4.0.2
  • sphinx-copybutton *
  • sphinx_markdown_tables >=0.0.16
  • sphinx_rtd_theme ==0.5.2
requirements/mminstall.txt pypi
  • mmcv-full >=1.5.0
requirements/optional.txt pypi
  • imagecorruptions *
  • scikit-learn *
  • scipy *
requirements/readthedocs.txt pypi
  • e2cnn *
  • mmcv *
  • mmdet >=2.25.1,<3.0.0
  • torch *
  • torchvision *
requirements/runtime.txt pypi
  • matplotlib *
  • mmcv-full *
  • mmdet >=2.25.1,<3.0.0
  • numpy *
  • pycocotools *
  • six *
  • terminaltables *
  • torch *
requirements/tests.txt pypi
  • asynctest * test
  • codecov * test
  • coverage * test
  • cython * test
  • flake8 * test
  • interrogate * test
  • isort ==4.3.21 test
  • kwarray * test
  • matplotlib * test
  • pytest * test
  • scikit-learn * test
  • ubelt * test
  • wheel * test
  • xdoctest >=0.10.0 test
  • yapf * test
requirements.txt pypi
setup.py pypi