-
Notifications
You must be signed in to change notification settings - Fork 3
/
test.sh
executable file
·36 lines (28 loc) · 944 Bytes
/
test.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#!/usr/bin/env bash
SCRIPTPATH="$( cd "$(dirname "$0")" ; pwd -P )"
SEGMENTATION_FILE="/output/images/breast-cancer-segmentation-for-tils/segmentation.tif"
DETECTION_FILE="/output/detected-lymphocytes.json"
TILS_SCORE_FILE="/output/til-score.json"
MEMORY=30g
echo "Building docker"
./build.sh
echo "Removing volume..."
docker volume rm tiager-ensemble
docker volume rm tiager-ensemble-tmp
echo "Creating volume..."
docker volume create tiager-ensemble
docker volume create tiager-ensemble-tmp
echo $SCRIPTPATH/testinput/
echo "Running algorithm..."
docker run --rm \
--memory=$MEMORY \
--memory-swap=$MEMORY \
--network=none \
--cap-drop=ALL \
--security-opt="no-new-privileges" \
--shm-size=128m \
--pids-limit=256 \
-v tiager-ensemble:/output/ \
-v tiager-ensemble-tmp:/tempoutput/ \
--runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=0 \
tiagerensemble