Data

Tools

Indexes

Applications

Experiments

Open Source Science

mmdetection-dinov2

https://github.com/oghanst/mmdetection-dinov2

Science Score: 44.0%

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

  • ✓
    CITATION.cff file
    Found 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 (2.0%) to scientific vocabulary
Last synced: 7 months ago · JSON representation ·

Repository

Basic Info
  • Host: GitHub
  • Owner: Oghanst
  • License: apache-2.0
  • Language: Python
  • Default Branch: main
  • Size: 61.7 MB
Statistics
  • Stars: 1
  • Watchers: 1
  • Forks: 0
  • Open Issues: 0
  • Releases: 0
Created about 2 years ago · Last pushed about 2 years ago
Metadata Files
Readme Contributing License Code of conduct Citation

README.md

开始

本项目主要基于mmengine-template实现 DINO-DINOv2模型

安装

请首先安装 pytorch1.12.1，cuda11.3和 cudnn8-devel。 git clone https://github.com/Oghanst/mmdetection-dinov2.git cd mmdetection-dinov2.git pip install -r requirements.txt

训练

  1. 设置数据集：

在路径 ./configs/_base_/datasets 创建自己的数据集配置文件，具体可以参考文件coco_detection_visdrone的设置。

对于里面的dataset_type = 'CocoDataset'需要在./mmdet/datasets/coco.py设计一个自己的新类，可以参考里面的class CocoDataset_Smokefire(CocoDataset):的设置。

然后在datasets的init.py文件里面对自己设计的新类进行注册。

  1. 设置PYTHONPATH export PYTHONPATH = $PWD:$PYTHONPATH

  2. 如何训练 参考configs里面的注释输入命令行./configs/dino python tools/train.py configs/dino/yourConfigName.py --work-dir work_dirs/dino

Owner

  • Name: Jiang Haojun
  • Login: Oghanst
  • Kind: user
  • Location: Minhang Qu, Shanghai, China
  • Company: Shanghai Jiao Tong University

I am a graduate student persuing master's degree at SJTU. I specialize in Computer Science and I am passionate about Artificial Intelligence.

Citation (CITATION.cff) 

cff-version: 1.2.0
message: "If you use this software, please cite it as below."
authors:
  - name: "MMDetection Contributors"
title: "OpenMMLab Detection Toolbox and Benchmark"
date-released: 2018-08-22
url: "https://github.com/open-mmlab/mmdetection"
license: Apache-2.0

GitHub Events

Total
Last Year

Dependencies

.github/workflows/deploy.yml actions
  • actions/checkout v2 composite
  • actions/setup-python v2 composite
.circleci/docker/Dockerfile docker
  • pytorch/pytorch ${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel build
.devcontainer/Dockerfile docker
  • pytorch/pytorch 2.1.2-cuda11.8-cudnn8-devel build
.devcontainer/docker-compose.yml docker
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
docker/serve_cn/Dockerfile docker
  • pytorch/pytorch ${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel build
requirements/albu.txt pypi
  • albumentations >=0.3.2
requirements/build.txt pypi
  • cython *
  • numpy *
requirements/docs.txt pypi
  • docutils ==0.16.0
  • myst-parser *
  • sphinx ==4.0.2
  • sphinx-copybutton *
  • sphinx_markdown_tables *
  • sphinx_rtd_theme ==0.5.2
  • urllib3 <2.0.0
requirements/mminstall.txt pypi
  • mmcv >=2.0.0rc4,<2.2.0
  • mmengine >=0.7.1,<1.0.0
requirements/multimodal.txt pypi
  • fairscale *
  • nltk *
  • pycocoevalcap *
  • transformers *
requirements/optional.txt pypi
  • cityscapesscripts *
  • fairscale *
  • imagecorruptions *
  • scikit-learn *
requirements/readthedocs.txt pypi
  • mmcv >=2.0.0rc4,<2.2.0
  • mmengine >=0.7.1,<1.0.0
  • scipy *
  • torch *
  • torchvision *
  • urllib3 <2.0.0
requirements/runtime.txt pypi
  • matplotlib *
  • numpy *
  • pycocotools *
  • scipy *
  • shapely *
  • six *
  • terminaltables *
  • tqdm *
requirements/tests.txt pypi
  • asynctest * test
  • cityscapesscripts * test
  • codecov * test
  • flake8 * test
  • imagecorruptions * test
  • instaboostfast * test
  • interrogate * test
  • isort ==4.3.21 test
  • kwarray * test
  • memory_profiler * test
  • nltk * test
  • onnx ==1.7.0 test
  • onnxruntime >=1.8.0 test
  • parameterized * test
  • prettytable * test
  • protobuf <=3.20.1 test
  • psutil * test
  • pytest * test
  • transformers * test
  • ubelt * test
  • xdoctest >=0.10.0 test
  • yapf * test
requirements/tracking.txt pypi
  • mmpretrain *
  • motmetrics *
  • numpy <1.24.0
  • scikit-learn *
  • seaborn *
requirements.txt pypi
  • GitPython ==3.1.40
  • Markdown ==3.5.1
  • addict ==2.4.0
  • aliyun-python-sdk-core ==2.14.0
  • aliyun-python-sdk-kms ==2.16.2
  • appdirs ==1.4.4
  • blessed ==1.20.0
  • click ==8.1.7
  • colorama ==0.4.6
  • contourpy ==1.1.1
  • crcmod ==1.7
  • cycler ==0.12.1
  • docker-pycreds ==0.4.0
  • einops ==0.7.0
  • fonttools ==4.47.0
  • fsspec ==2023.12.2
  • gitdb ==4.0.11
  • gpustat ==1.1.1
  • huggingface-hub ==0.20.1
  • importlib-metadata ==7.0.1
  • importlib-resources ==6.1.1
  • jmespath ==0.10.0
  • kiwisolver ==1.4.5
  • markdown-it-py ==3.0.0
  • mat4py ==0.6.0
  • matplotlib ==3.7.4
  • mdurl ==0.1.2
  • mkl-service ==2.4.0
  • mmcv ==2.1.0
  • mmengine ==0.10.2
  • mmpretrain ==1.1.1
  • model-index ==0.1.11
  • modelindex ==0.0.2
  • nvidia-ml-py ==12.535.133
  • opencv-python ==4.8.1.78
  • opendatalab ==0.0.10
  • openmim ==0.3.9
  • openxlab ==0.0.32
  • ordered-set ==4.1.0
  • oss2 ==2.17.0
  • pandas ==2.0.3
  • protobuf ==4.25.1
  • pycocotools ==2.0.7
  • pycryptodome ==3.19.0
  • pyparsing ==3.1.1
  • pytz ==2023.3.post1
  • requests ==2.28.2
  • rich ==13.4.2
  • safetensors ==0.4.1
  • scipy ==1.10.1
  • sentry-sdk ==1.39.1
  • setproctitle ==1.3.3
  • shapely ==2.0.2
  • smmap ==5.0.1
  • tabulate ==0.9.0
  • termcolor ==2.4.0
  • terminaltables ==3.1.10
  • timm ==0.9.12
  • tomli ==2.0.1
  • torch ==2.1.2
  • torchaudio ==2.1.2
  • torchvision ==0.16.1
  • tqdm ==4.65.2
  • triton ==2.1.0
  • tzdata ==2023.3
  • wandb ==0.16.1
  • yapf ==0.40.2
  • zipp ==3.17.0
setup.py pypi