Commit 7d06b2f4 authored by Huiying Zhang's avatar Huiying Zhang
Browse files

Update dcnn.py

parent 14ee3463
......@@ -29,6 +29,7 @@ def extract_data_labels(filename):
img_ang = row["img_ang"]
ids.append(id_)
data.append(np.stack((img_abs, img_ang), 2))
#此处应该是因为ID只有三个所以才是【0,0,0】的状态,之后要改
one_hot_label = [0, 0, 0]
one_hot_label[label] = 1
labels.append(one_hot_label)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment