wassname/CoordConv
1
0
Fork 0
mirror of https://github.com/wassname/CoordConv.git synced 2026-09-11 11:52:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

add quadrant dataset predictions plot

Browse Source
This commit is contained in:
walsvid
2018-07-16 21:09:18 +08:00
parent 0fc808eaf4
commit bd5fd553b8
3 changed files with 46 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
experiments/generate_data.py
+3 -3
View File
@@ -67,8 +67,8 @@ else:
train_set = np.array(train_set)
test_set = np.array(test_set)
train_set = train_set[:, None, None, :]
test_set = test_set[:, None, None, :]
train_set = train_set[:, :, None, None]
test_set = test_set[:, :, None, None]
print(train_set.shape)
print(test_set.shape)
@@ -87,4 +87,4 @@ else:
np.save('data-quadrant/train_onehot.npy', train_onehot)
np.save('data-quadrant/train_images.npy', train_images)
np.save('data-quadrant/test_onehot.npy', test_onehot)
np.save('data-quadrant/test_images.npy', test_images)
np.save('data-quadrant/test_images.npy', test_images)