-
Notifications
You must be signed in to change notification settings - Fork 4
/
trainsetup.jl
39 lines (36 loc) · 1.77 KB
/
trainsetup.jl
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
using Pkg; Pkg.activate("."); Pkg.instantiate(); #install required packages
server = "ai.ku.edu.tr/"
if length(ARGS)==0
println("clevr home folder is not specified. Pre-processed data will be downloaded from the servers(70GB) if they are not in the data folder? (yes or no)")
if readline() == "yes"
!isfile("data/train.bin") && download(server*"data/mac-network/train.bin","data/train.bin")
!isfile("data/val.bin") && download(server*"data/mac-network/val.bin","data/val.bin")
!isfile("data/train.json") && download(server*"data/mac-network/train.json","data/train.json")
!isfile("data/val.json") && download(server*"data/mac-network/val.json","data/val.json")
!isfile("data/dic.json") && download(server*"data/mac-network/dic.json","data/dic.json")
exit()
else
error("No data available!")
end
else
CLEVR_HOME = ARGS[1]
println(CLEVR_HOME)
end
include("preprocess.jl")
println("Starting to preprocess CLEVR questions @: $(CLEVR_HOME)/questions ")
println("Training and validation questions will be processed...")
if isfile("data/train.json") && isfile("data/val.json")
println("You already had processed questions in your data folder. Do you want to re-process the data? (yes or no)")
readline() == "yes" && preprocess([CLEVR_HOME])
else
preprocess([CLEVR_HOME])
end
include("extract_features.jl")
println("Starting to preprocess CLEVR images @: $(CLEVR_HOME)/images ")
println("Training and validation images will be processed...")
if isfile("data/train.bin") && isfile("data/val.bin")
println("You already had processed images in your data folder. Do you want to re-process the data? (yes or no)")
readline() == "yes" && extract_features([CLEVR_HOME])
else
extract_features([CLEVR_HOME])
end