From 9566a42548f2a8018715307ddbb8300ffec3128f Mon Sep 17 00:00:00 2001
From: Shivam Khandelwal <shivam@aicrowd.com>
Date: Fri, 28 May 2021 02:42:40 +0000
Subject: [PATCH] SiamMot orchestration
---
.gitignore | 1 -
siam-mot/Dockerfile | 40 ++++++++++++++++++++++++++++
siam-mot/requirements_additional.txt | 2 ++
siam-mot/requirements_exact.txt | 16 +++++++++++
4 files changed, 58 insertions(+), 1 deletion(-)
create mode 100644 siam-mot/Dockerfile
create mode 100644 siam-mot/requirements_additional.txt
create mode 100644 siam-mot/requirements_exact.txt
diff --git a/.gitignore b/.gitignore
index bedb587..fa61ab2 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 0000000..dd0da94
--- /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 0000000..54bc657
--- /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 0000000..e35df66
--- /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
--
GitLab