Add a script and reqs to run XSeg editor

2021-08-01 17:52:35 -07:00 · 2021-08-01 17:52:35 -07:00 · 7dcb9568c9
parent 6bf4d54800
commit 7dcb9568c9
3 changed files with 8 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -8,3 +8,4 @@ ffmpeg-python==0.2.0
 Pillow==7.2.0
 scikit-image==0.17.2
 h5py==3.1.0
+PyQt5==5.15.4
--- a/requirements_3.6.txt
+++ b/requirements_3.6.txt
@ -8,3 +8,4 @@ ffmpeg-python==0.2.0
 Pillow==7.2.0
 scikit-image==0.17.2
 h5py==2.10.0
+PyQt5==5.15.4
--- a/scripts/xseg.sh
+++ b/scripts/xseg.sh
@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" xseg editor \
+  --input-dir "$WORKSPACE/data_src/aligned"