VSAG/vision_sjtu_19
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新CNN参数

Browse Source
This commit is contained in:
xinyang
2019-05-05 18:37:30 +08:00
parent 479e43d2af
commit c09f0d3a2d
10 changed files with 19911 additions and 12337 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tools/TrainCNN/backward.py
View File

@@ -54,7 +54,7 @@ def save_para(folder, paras):
save_bias(fp, paras[7])
STEPS = 20000
STEPS = 100000
BATCH = 30
LEARNING_RATE_BASE = 0.01
LEARNING_RATE_DECAY = 0.99
@@ -152,7 +152,7 @@ def train(dataset, show_bar=False):
if __name__ == "__main__":
print("Loading data sets...")
dataset = generate.DataSet("/home/xinyang/Desktop/DataSets/box")
dataset = generate.DataSet("/home/xinyang/Desktop/DataSets")
print("Finish!")
train(dataset, show_bar=True)
input("Press any key to end...")