pyro-train/Makefile at main · pyronear/pyro-train · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
.PHONY: check fix mlflow_start mlflow_stop run_yolo_wide_hyperparameter_search run_yolo_narrow_hyperparameter_search run_yolo_benchmark run_test_suite

check:
	uv run isort --check .
	uv run flake8 .
	uv run mypy .
	uv run black --check .

fix:
	uv run isort .
	uv run black .

mlflow_start:
	uv run mlflow server --backend-store-uri runs/mlflow

mlflow_stop:
	ps aux | grep 'mlflow' | grep -v 'grep' | awk '{print $2}' | xargs kill -9

run_yolo_wide_hyperparameter_search:
	uv run python ./scripts/model/yolo/hyperparameter_search.py \
	  --data ./data/03_model_input/wildfire/small/datasets/data.yaml \
	  --output-dir ./data/04_models/yolo/ \
	  --experiment-name "random_hyperparameter_search" \
	  --filepath-space-yaml ./scripts/model/yolo/spaces/wide.yaml \
	  --n 50 \
	  --loglevel "info"

run_yolo_narrow_hyperparameter_search:
	uv run python ./scripts/model/yolo/hyperparameter_search.py \
	  --data ./data/03_model_input/wildfire/full/datasets/data.yaml \
	  --output-dir ./data/04_models/yolo/ \
	  --experiment-name "random_hyperparameter_search" \
	  --filepath-space-yaml ./scripts/model/yolo/spaces/narrow.yaml \
	  --n 5 \
	  --loglevel "info"

run_yolo_benchmark:
	uv run python ./scripts/model/yolo/benchmark.py \
	  --input-dir ./data/04_models/yolo/ \
	  --output-dir ./data/06_reporting/yolo/ \
	  --loglevel "info"

run_test_suite:
	uv run pytest