-
Notifications
You must be signed in to change notification settings - Fork 8
/
Makefile
228 lines (204 loc) · 6.77 KB
/
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
.PHONY: clean data lint requirements sync_data_to_s3 sync_data_from_s3
#################################################################################
# GLOBALS #
#################################################################################
PROJECT_DIR := $(shell dirname $(realpath $(lastword $(MAKEFILE_LIST))))
BUCKET = [OPTIONAL] your-bucket-for-syncing-data (do not include 's3://')
PROFILE = default
PROJECT_NAME = fusion
PYTHON_INTERPRETER = python3
# Make Data variables
DATA_PATH:="$(shell pwd)/data/raw/"
PROCESSED_DATA_PATH:="$(shell pwd)/data/processed/"
DATASET_TYPE = "" # [SKELETON, RGB, IR]
COMPRESSION = ""
COMPRESSION_OPTS = 9
# Make Train variables
MODEL_FOLDER :="$(shell pwd)/models/"
EVALUATION_TYPE=cross_subject
MODEL_TYPE=FUSION
USE_POSE=False
USE_IR=False
PRETRAINED=False
USE_CROPPED_IR=False
FUSION_SCHEME=CONCAT
OPTIMIZER=ADAM
LEARNING_RATE=1e-4
WEIGHT_DECAY=0
GRADIENT_THRESHOLD=0
EPOCHS=30
BATCH_SIZE=8
ACCUMULATION_STEPS=1
SUB_SEQUENCE_LENGTH=20
AUGMENT_DATA=True
MIRROR_SKELETON=False
EVALUATE_TEST=True
SEED=0
# Plot confusion matrix variables
MODEL_FILE=""
CPKL_FILE=""
ifeq (,$(shell which conda))
HAS_CONDA=False
else
HAS_CONDA=True
endif
#################################################################################
# COMMANDS #
#################################################################################
## Install Python Dependencies
requirements: test_environment
$(PYTHON_INTERPRETER) -m pip install -U pip setuptools wheel
$(PYTHON_INTERPRETER) -m pip install -r requirements.txt
## Make h5 Dataset
data:
$(PYTHON_INTERPRETER) src/data/make_dataset.py \
--data_path=$(DATA_PATH) \
--output_folder=$(PROCESSED_DATA_PATH) \
--dataset_type=$(DATASET_TYPE) \
--compression=$(COMPRESSION) \
--compression_opts=$(COMPRESSION_OPTS)
## Make Train
train:
$(PYTHON_INTERPRETER) src/models/train_model.py \
--data_path=$(PROCESSED_DATA_PATH) \
--output_folder=$(MODEL_FOLDER) \
--evaluation_type=$(EVALUATION_TYPE) \
--model_type=$(MODEL_TYPE) \
--use_pose=$(USE_POSE) \
--use_ir=$(USE_IR) \
--pretrained=$(PRETRAINED) \
--use_cropped_IR=$(USE_CROPPED_IR) \
--fusion_scheme=$(FUSION_SCHEME) \
--optimizer=$(OPTIMIZER) \
--learning_rate=$(LEARNING_RATE) \
--weight_decay=$(WEIGHT_DECAY) \
--gradient_threshold=$(GRADIENT_THRESHOLD) \
--epochs=$(EPOCHS) \
--batch_size=$(BATCH_SIZE) \
--accumulation_steps=$(ACCUMULATION_STEPS) \
--sub_sequence_length=$(SUB_SEQUENCE_LENGTH) \
--augment_data=$(AUGMENT_DATA) \
--mirror_skeleton=$(MIRROR_SKELETON) \
--evaluate_test=$(EVALUATE_TEST) \
--seed=$(SEED)
## Make confusion matrix from model
confusion_matrix:
$(PYTHON_INTERPRETER) src/models/plot_confusion_matrix.py \
--data_path=$(PROCESSED_DATA_PATH) \
--model_folder=$(MODEL_FOLDER) \
--model_file=$(MODEL_FILE) \
--evaluation_type=$(EVALUATION_TYPE) \
--model_type=$(MODEL_TYPE) \
--use_pose=$(USE_POSE) \
--use_ir=$(USE_IR) \
--fusion_scheme=$(FUSION_SCHEME) \
--use_cropped_IR=$(USE_CROPPED_IR) \
--batch_size=$(BATCH_SIZE) \
--sub_sequence_length=$(SUB_SEQUENCE_LENGTH)
## Make confusion matrix from cpkl file
confusion_matrix_cpkl:
$(PYTHON_INTERPRETER) src/models/plot_confusion_matrix_cpkl.py \
--model_folder=$(MODEL_FOLDER) \
--cpkl_file=$(CPKL_FILE)
## Delete all compiled Python files
clean:
find . -type f -name "*.py[co]" -delete
find . -type d -name "__pycache__" -delete
## Lint using flake8
lint:
flake8 src
## Upload Data to S3
sync_data_to_s3:
ifeq (default,$(PROFILE))
aws s3 sync data/ s3://$(BUCKET)/data/
else
aws s3 sync data/ s3://$(BUCKET)/data/ --profile $(PROFILE)
endif
## Download Data from S3
sync_data_from_s3:
ifeq (default,$(PROFILE))
aws s3 sync s3://$(BUCKET)/data/ data/
else
aws s3 sync s3://$(BUCKET)/data/ data/ --profile $(PROFILE)
endif
## Set up python interpreter environment
create_environment:
ifeq (True,$(HAS_CONDA))
@echo ">>> Detected conda, creating conda environment."
ifeq (3,$(findstring 3,$(PYTHON_INTERPRETER)))
conda create --name $(PROJECT_NAME) python=3
else
conda create --name $(PROJECT_NAME) python=2.7
endif
@echo ">>> New conda env created. Activate with:\nsource activate $(PROJECT_NAME)"
else
$(PYTHON_INTERPRETER) -m pip install -q virtualenv virtualenvwrapper
@echo ">>> Installing virtualenvwrapper if not already intalled.\nMake sure the following lines are in shell startup file\n\
export WORKON_HOME=$$HOME/.virtualenvs\nexport PROJECT_HOME=$$HOME/Devel\nsource /usr/local/bin/virtualenvwrapper.sh\n"
@bash -c "source `which virtualenvwrapper.sh`;mkvirtualenv $(PROJECT_NAME) --python=$(PYTHON_INTERPRETER)"
@echo ">>> New virtualenv created. Activate with:\nworkon $(PROJECT_NAME)"
endif
## Test python environment is setup correctly
test_environment:
$(PYTHON_INTERPRETER) test_environment.py
#################################################################################
# PROJECT RULES #
#################################################################################
#################################################################################
# Self Documenting Commands #
#################################################################################
.DEFAULT_GOAL := help
# Inspired by <http://marmelab.com/blog/2016/02/29/auto-documented-makefile.html>
# sed script explained:
# /^##/:
# * save line in hold space
# * purge line
# * Loop:
# * append newline + line to hold space
# * go to next line
# * if line starts with doc comment, strip comment character off and loop
# * remove target prerequisites
# * append hold space (+ newline) to line
# * replace newline plus comments by `---`
# * print line
# Separate expressions are necessary because labels cannot be delimited by
# semicolon; see <http://stackoverflow.com/a/11799865/1968>
.PHONY: help
help:
@echo "$$(tput bold)Available rules:$$(tput sgr0)"
@echo
@sed -n -e "/^## / { \
h; \
s/.*//; \
:doc" \
-e "H; \
n; \
s/^## //; \
t doc" \
-e "s/:.*//; \
G; \
s/\\n## /---/; \
s/\\n/ /g; \
p; \
}" ${MAKEFILE_LIST} \
| LC_ALL='C' sort --ignore-case \
| awk -F '---' \
-v ncol=$$(tput cols) \
-v indent=19 \
-v col_on="$$(tput setaf 6)" \
-v col_off="$$(tput sgr0)" \
'{ \
printf "%s%*s%s ", col_on, -indent, $$1, col_off; \
n = split($$2, words, " "); \
line_length = ncol - indent; \
for (i = 1; i <= n; i++) { \
line_length -= length(words[i]) + 1; \
if (line_length <= 0) { \
line_length = ncol - indent - length(words[i]) - 1; \
printf "\n%*s ", -indent, " "; \
} \
printf "%s ", words[i]; \
} \
printf "\n"; \
}' \
| more $(shell test $(shell uname) = Darwin && echo '--no-init --raw-control-chars')