Skip to content

Commit e8083c9

Browse files
author
Miles Xu
committed
dataset test
1 parent 62b0274 commit e8083c9

File tree

4 files changed

+170
-30
lines changed

4 files changed

+170
-30
lines changed
Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
{
2+
"cells": [
3+
{
4+
"cell_type": "code",
5+
"execution_count": null,
6+
"metadata": {},
7+
"outputs": [],
8+
"source": []
9+
}
10+
],
11+
"metadata": {
12+
"kernelspec": {
13+
"display_name": "Python 3",
14+
"language": "python",
15+
"name": "python3"
16+
},
17+
"language_info": {
18+
"codemirror_mode": {
19+
"name": "ipython",
20+
"version": 3
21+
},
22+
"file_extension": ".py",
23+
"mimetype": "text/x-python",
24+
"name": "python",
25+
"nbconvert_exporter": "python",
26+
"pygments_lexer": "ipython3",
27+
"version": "3.7.2"
28+
}
29+
},
30+
"nbformat": 4,
31+
"nbformat_minor": 2
32+
}

ground_data_explore.ipynb

Lines changed: 119 additions & 0 deletions
Large diffs are not rendered by default.

ground_visualization.ipynb

Lines changed: 16 additions & 27 deletions
Large diffs are not rendered by default.

lens_main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,12 @@
1111
from learning_rate_update import learning_rate_update
1212

1313
if sys.platform == 'linux':
14-
path = '/home/milesx/datasets/deeplens/'
14+
path = '/home/mingx/datasets/'
1515
else:
1616
path = 'C:\\Users\\miles\\Documents\\dataset'
1717
save_path = os.path.join(path, 'saved_model')
1818
train_params = {
19-
'n_data': 1024,
19+
'n_data': 5120,
2020
'num_classes': 2,
2121
'batch_size': 128,
2222
'n_eval_data': 128,
@@ -45,7 +45,7 @@
4545
torch.cuda.manual_seed_all(770715)
4646
torch.backends.cudnn.deterministic = True
4747
train_composed = transforms.Compose(
48-
[Log10(), Clamp(1e-9, 100), WhitenInput(),
48+
[WhitenInput(),
4949
AugmentTranslate(train_params['augment_translation'], 101)])
5050
test_composed = transforms.Compose([Clamp(1e-9, 100), WhitenInput()])
5151
ground_train_dataset = gbd.GroundBasedDataset(

0 commit comments

Comments
 (0)