integration_test.yml

name: DetectTrack

on:
  push:
    branches: [ main, action ]
  pull_request:
    branches: [ main, action ]

jobs:
  # This workflow contains a single job called "build"
  build:
    runs-on: ubuntu-latest

    steps:
      - uses: actions/checkout@v2

      - name: Set up Python 3.10
        uses: actions/setup-python@v2
        with:
          python-version: '3.10'
          architecture: 'x64'

      - name: apt-get
        run: sudo apt-get update -y && sudo apt-get install wget unzip ffmpeg -y

      - name: Install
        run: pip3 install --upgrade -r requirements.txt && pip3 install -e .

      - name: Small train
        run: python3 train.py --train_steps 3 --nframes=5 --kpoints=11 --npca=3 --size=64 --nworms_pca=25 --nworms=5,10 --batch_size=4 --clip_duration=0.2 --eval_interval=2 --latent_dim=2

      - name: Download video
        run: wget https://sid.erda.dk/share_redirect/GVDAVHOZLN -O video.mp4
     
      - name: Download weights
        run: wget https://sid.erda.dk/share_redirect/cEjIpG1yQl -O weights.zip && unzip weights.zip

      - name: Detect
        run: python3 examples/detect.py --model=weights/ --input=video.mp4 --output=output.png --frame=12 --correction_factor=1.2

      - name: Track
        run: python3 examples/track.py --model=weights/ --input=video.mp4 --output=track/ --correction_factor=1.2 --num_batches=1

      - name: Store artefacts
        uses: actions/upload-artifact@v3
        with:
          name: detect-track-output
          path: |
            output.png
            tracking.mp4