-
Notifications
You must be signed in to change notification settings - Fork 0
/
Makefile
44 lines (35 loc) · 815 Bytes
/
Makefile
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
SHELL := /bin/bash
.EXPORT_ALL_VARIABLES:
PYTHONPATH := ./
TEST_DIR := tests/
LINT_DIR := ./
lint:
flake8 ${LINT_DIR}
# Call this to format your code.
format:
isort .
black .
verify_format:
black --check .
isort --check .
run_tests:
pytest -svvv ${TEST_DIR}
reset_logs:
rm -rf logs
mkdir logs
rm -rf wandb
export_to_onnx:
python export_to_onnx.py \
--config config.yml \
--torch_weights logs/2023-04-12\ 08\:04\:42.694413/weights/fcn_8_iou_0.6438923478126526.pt \
--onnx_path ./fcn.onnx \
--image_size 500,500 \
--do_check_on_validation_set
predict:
python predict.py \
--image data/VOCdevkit/VOC2012/JPEGImages/2007_000363.jpg \
--config config.yml \
--weights weights/fcn_sim..onnx \
--image_size 500,500
# Call this before commit.
pre_push_test: verify_format lint run_tests