mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

Fix mmdet import and some typo

Browse Source
This commit is contained in:
triple-Mu 2022-11-08 16:19:39 +08:00
parent df962aa2a5
commit 7bf73b42d5
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
docker/GPU/Dockerfile
View File

@ -5,7 +5,6 @@ ARG PYTHON_VERSION=3.8
ARG TORCH_VERSION=1.10.0 ARG TORCH_VERSION=1.10.0
ARG TORCHVISION_VERSION=0.11.0 ARG TORCHVISION_VERSION=0.11.0
ARG ONNXRUNTIME_VERSION=1.8.1 ARG ONNXRUNTIME_VERSION=1.8.1
ARG MMCV_VERSION=1.5.3
ARG PPLCV_VERSION=0.7.0 ARG PPLCV_VERSION=0.7.0
ENV FORCE_CUDA="1" ENV FORCE_CUDA="1"
ARG MMCV_VERSION=">=2.0.0rc2,<2.1.0" ARG MMCV_VERSION=">=2.0.0rc2,<2.1.0"
@ -40,8 +39,8 @@ RUN curl -fsSL -v -o ~/miniconda.sh -O https://repo.anaconda.com/miniconda/Mini
RUN /opt/conda/bin/pip install --no-cache-dir openmim RUN /opt/conda/bin/pip install --no-cache-dir openmim
### pytorch mmcv onnxruntime ### pytorch mmcv onnxruntime
RUN mim install torch==${TORCH_VERSION}+cpu torchvision==${TORCHVISION_VERSION}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html \ RUN /opt/conda/bin/mim install torch==${TORCH_VERSION}+cpu torchvision==${TORCHVISION_VERSION}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html \
&& mim install --no-cache-dir "mmcv"${MMCV_VERSION} onnxruntime-gpu==${ONNXRUNTIME_VERSION} mmengine${MMENGINE_VERSION} && /opt/conda/bin/mim install --no-cache-dir "mmcv"${MMCV_VERSION} onnxruntime-gpu==${ONNXRUNTIME_VERSION} mmengine${MMENGINE_VERSION}
ENV PATH /opt/conda/bin:$PATH ENV PATH /opt/conda/bin:$PATH
@ -66,7 +65,7 @@ RUN git clone https://github.com/open-mmlab/mmdeploy &&\
cmake -DMMDEPLOY_TARGET_BACKENDS="ort;trt" .. &&\ cmake -DMMDEPLOY_TARGET_BACKENDS="ort;trt" .. &&\
make -j$(nproc) &&\ make -j$(nproc) &&\
cd .. &&\ cd .. &&\
mim install -e . /opt/conda/bin/mim install -e .
### build sdk ### build sdk
RUN git clone https://github.com/openppl-public/ppl.cv.git &&\ RUN git clone https://github.com/openppl-public/ppl.cv.git &&\

2
mmdeploy/codebase/mmdet/models/dense_heads/yolo_head.py
View File

@ -3,7 +3,7 @@ from typing import Sequence
import numpy as np import numpy as np
import torch import torch
from mmdet.utils.typing import OptConfigType from mmdet.utils import OptConfigType
from torch import Tensor from torch import Tensor
from mmdeploy.codebase.mmdet.deploy import (get_post_processing_params, from mmdeploy.codebase.mmdet.deploy import (get_post_processing_params,