diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index ee8f2a7f14f35b690dda9c285de411bf1f759a4f..f40632d57328c57a8aa5988ce3b95246becefa49 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -57,7 +57,19 @@ image: # TAG: 0119cv6 # RECIPE_PATH: recipes/quality-control/fastqc/ +##################### +# DeepLabCut +##################### +deeplabcut-2202-builddeploy: + extends: .templateBuildDeploy + variables: + BASENAME: deeplabcut + TAG: 2.2.0.2 + RECIPE_PATH: recipes/deeplabcut/ + only: + changes: + - recipes/deeplabcut/Singularity.deeplabcut-2202 ##################### ## Quality control diff --git a/recipes/deeplabcut/Singularity.deeplabcut-2202 b/recipes/deeplabcut/Singularity.deeplabcut-2202 new file mode 100644 index 0000000000000000000000000000000000000000..0aeb1e6b73e3c367f798f60cdb7e449f849e6e2c --- /dev/null +++ b/recipes/deeplabcut/Singularity.deeplabcut-2202 @@ -0,0 +1,26 @@ +Bootstrap: docker +From: deeplabcut/deeplabcut:latest-core + +%help + A Singularity image from the deeplabcut:latest-core Docker. It comes with ResNet weights embedded. + +%labels + AUTHOR Francesco Tabaro + VERSION 0.1 + +%post + apt-get update && apt-get install -y curl + + cd /usr/local/lib/python3.8/dist-packages/deeplabcut/pose_estimation_tensorflow/models/pretrained/ + + # Adapted from download.sh + for i in resnet_v1_50_2016_08_28.tar.gz resnet_v1_101_2016_08_28.tar.gz resnet_v1_152_2016_08_28.tar.gz; do + curl -O http://download.tensorflow.org/models/$i + FN=$(tar xzvf $i) + chmod 666 $FN + rm $i + done + +%runscript + exec /usr/bin/python3 "$@" +