diff --git a/.gitignore b/.gitignore index bedb5871a53a9dfee919ffe3e5556f1ba622fe29..fa61ab28c771ffa534150dcaaafb3e6562912a3c 100644 --- a/.gitignore +++ b/.gitignore @@ -49,7 +49,6 @@ htmlcov/ .nox/ .coverage .coverage.* -.cache nosetests.xml coverage.xml *.cover diff --git a/siam-mot/Dockerfile b/siam-mot/Dockerfile new file mode 100644 index 0000000000000000000000000000000000000000..dd0da945c45d85abf7024562ccc681f061599702 --- /dev/null +++ b/siam-mot/Dockerfile @@ -0,0 +1,40 @@ +FROM aicrowd/base-images:py37-cuda11-torch171-maskrcnn + +# Set up AIcrowd user for evaluation +ARG NB_USER=aicrowd +ARG NB_UID=1001 +ENV USER ${NB_USER} +ENV HOME /home/${NB_USER} + +RUN adduser --disabled-password \ + --gecos "Default user" \ + --uid ${NB_UID} \ + ${NB_USER} + +ARG REPO_DIR=${HOME} +ENV REPO_DIR ${REPO_DIR} +WORKDIR ${REPO_DIR} + +# Install additional dependencies +COPY siam-mot/requirements_exact.txt requirements_exact.txt +RUN pip install -r requirements_exact.txt --use-deprecated=legacy-resolver + +COPY siam-mot/requirements_additional.txt requirements_additional.txt +RUN pip install -r requirements_additional.txt + +COPY requirements.txt requirements.txt +RUN pip install -r requirements.txt + +# Add siam-mot related PATH +ENV PATH ${HOME}/.local/bin:${REPO_DIR}/.local/bin:${PATH} +ENV PATH="${PATH}:/home/aicrowd/siam-mot" +ENV PYTHONPATH="${PYTHONPATH}:/home/aicrowd/siam-mot" + +# Copy current directory as submission +COPY . ${REPO_DIR} + +# Entry point for evaluation +# You can skip in case you are using this Dockerfile for another purpose +RUN chown -R ${NB_USER}:${NB_USER} ${REPO_DIR} +USER ${NB_USER} +ENTRYPOINT ["/home/aicrowd/run.sh"] diff --git a/siam-mot/requirements_additional.txt b/siam-mot/requirements_additional.txt new file mode 100644 index 0000000000000000000000000000000000000000..54bc65710a6f8b70751f7d25d8de6985ddcc4c72 --- /dev/null +++ b/siam-mot/requirements_additional.txt @@ -0,0 +1,2 @@ +cityscapesscripts +ffmpeg diff --git a/siam-mot/requirements_exact.txt b/siam-mot/requirements_exact.txt new file mode 100644 index 0000000000000000000000000000000000000000..e35df666fd7eaf0608cb34a5770b64f6f819f695 --- /dev/null +++ b/siam-mot/requirements_exact.txt @@ -0,0 +1,16 @@ +numpy==1.20.1 +gluoncv==0.10.0 +motmetrics==1.2.0 +timm==0.4.5 +tqdm==4.59.0 +imgaug==0.4.0 +# torchvision==0.8.2 +# torch==1.7.1 +yacs==0.1.8 +Pillow==8.2.0 +pycocotools==2.0.2 +scikit_learn==0.24.2 +indexed==1.2.0 +fire==0.4.0 +mxnet==1.8.0.post0 +tensorboard==2.5.0