Merge pull request #15 from kirkegaardlab/fix_jax_deprecation #21

	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

Provide feedback