openiss-yolo-gpu.sh

#!/encs/bin/tcsh

# Give job a name
#SBATCH -J oi-yolo-gpu

# Set output directory to current
#SBATCH --chdir=./

# Send an email when the job starts, finishes or if it is aborted.
#SBATCH --mail-type=ALL

# Request GPU
#SBATCH --gpus=2

# Request CPU with maximum memoy size = 40GB
#SBATCH --mem=40G

# Request CPU slots 
#SBATCH -n 16

#sleep 30

# Specify the output file name
#SBATCH -o openiss-yolo-batch-gpu.log

module load anaconda3/2023.03/default

conda activate /speed-scratch/$USER/YOLO

# Image example 
#srun python yolo_video.py --model model_data/yolo.h5 --classes model_data/coco_classes.txt --image  --gpu_num 2

# Video example 
srun python yolo_video.py --input video/v1.avi --output video/002.avi --gpu_num 2

conda deactivate

# EOF