diff --git a/datasets/linemod/dataset.py b/datasets/linemod/dataset.py index f060ac310cee16a0ed6b39a0306659d57f484e12..86d7a8f353874c30ec74567ee941386fdf8a580b 100755 --- a/datasets/linemod/dataset.py +++ b/datasets/linemod/dataset.py @@ -61,7 +61,7 @@ class PoseDataset(data.Dataset): self.list_rgb.append('{0}/data/{1}/rgb/{2}.png'.format(self.root, '%02d' % item, input_line)) self.list_depth.append('{0}/data/{1}/depth/{2}.png'.format(self.root, '%02d' % item, input_line)) if self.mode == 'eval': - self.list_label.append('{0}/segnet_results_gt/{1}_label/{2}.png'.format(self.root, '%02d' % item, input_line)) + self.list_label.append('{0}/segnet_results/{1}_label_gt/{2}.png'.format(self.root, '%02d' % item, input_line)) else: self.list_label.append('{0}/data/{1}/mask/{2}.png'.format(self.root, '%02d' % item, input_line)) diff --git a/slurm/DF_eval.slurm b/slurm/DF_eval.slurm new file mode 100644 index 0000000000000000000000000000000000000000..8263796ec5071be5607c117552d23b5b4c75359c --- /dev/null +++ b/slurm/DF_eval.slurm @@ -0,0 +1,27 @@ +#!/bin/bash +#SBATCH --job-name=DF07cam # nom du job +#SBATCH -A uli@gpu +#SBATCH --ntasks=1 # Nombre de processus MPI +#SBATCH --cpus-per-task=10 # nombre de threads OpenMP +#SBATCH --hint=nomultithread # pas d'hyperthreading +#SBATCH --qos=qos_gpu-t4 +#SBATCH --gres=gpu:1 +#SBATCH --time=100:00:00 # Temps d’exécution max +#SBATCH --output=Output/Pvnet%j.out # fichier de sortie +#SBATCH --error=Output/Pvnet%j.out # fichier d'erreur + +# on se place dans le répertoire de soumission +cd ${SLURM_SUBMIT_DIR} + +module purge +# chargement des modules + +module load python/3.6.15 + +# echo des commandes lancées +set -x + + +conda activate NewDF +bash ./experiments/scripts/eval_linemod.sh + diff --git a/slurm/DF_train.slurm b/slurm/DF_train.slurm new file mode 100644 index 0000000000000000000000000000000000000000..d43ec06675a8c6371bf7c4c0e823b6427ef7bf8e --- /dev/null +++ b/slurm/DF_train.slurm @@ -0,0 +1,31 @@ +#!/bin/bash +#SBATCH --job-name=DF07cam # nom du job +#SBATCH -A uli@gpu +#SBATCH --ntasks=1 # Nombre de processus MPI +#SBATCH --cpus-per-task=10 # nombre de threads OpenMP +#SBATCH --hint=nomultithread # pas d'hyperthreading +#SBATCH --qos=qos_gpu-t4 +#SBATCH --gres=gpu:1 +#SBATCH --time=100:00:00 # Temps d’exécution max +#SBATCH --output=Output/Pvnet%j.out # fichier de sortie +#SBATCH --error=Output/Pvnet%j.out # fichier d'erreur + +# on se place dans le répertoire de soumission +cd ${SLURM_SUBMIT_DIR} + +module purge +# chargement des modules + +module load python/3.6.15 + +# echo des commandes lancées +set -x + + +cat_target=$1 +echo $cat_target + + +conda activate NewDF +bash ./experiments/scripts/train_linemod.sh +