Makefile

help:  ## Show help
	@grep -E '^[.a-zA-Z_-]+:.*?## .*$$' $(MAKEFILE_LIST) | awk 'BEGIN {FS = ":.*?## "}; {printf "\033[36m%-30s\033[0m %s\n", $$1, $$2}'

clean: ## Clean autogenerated files
	rm -rf dist
	find . -type f -name "*.DS_Store" -ls -delete
	find . | grep -E "(__pycache__|\.pyc|\.pyo)" | xargs rm -rf
	find . | grep -E ".pytest_cache" | xargs rm -rf
	find . | grep -E ".ipynb_checkpoints" | xargs rm -rf
	rm -f .coverage

clean-logs: ## Clean logs
	rm -rf logs/**
 
format: ## Run pre-commit hooks
	pre-commit run -a

sync: ## Merge changes from main branch to your current branch
	git pull
	git pull origin main

test: ## Run not slow tests
	pytest -k "not slow"

test-full: ## Run all tests
	pytest

train-yolo: ## Train YOLO model
	python src/train.py \
		model=yolo_nano \
		trainer=gpu \
		data.batch_size=2 \
		trainer.fast_dev_run=True
eval-yolo:
	python src/eval.py \
		model=yolo_nano \
		trainer=gpu \
		data.batch_size=2 \
		ckpt_path="./checkpoints/yolo_nano/epoch\=097.ckpt"

submit-yolo:
	python src/submit.py \
		model=old_yolo_nano \
		trainer=gpu \
		data=public_pixta_face \
		ckpt_path="./checkpoints/yolo_nano/last.ckpt"

export-yolo:
	python src/export.py \
		model=yolo_nano_v1 \
		ckpt_path="./checkpoints/yolo_nano_v1/epoch\=192.ckpt" \
		onnx_path="./checkpoints/yolo_nano_v1/yolo_nano_v1.onnx" \
		quantized_onnx_path="./checkpoints/yolo_nano_v1/yolo_nano_v1.quant.onnx"