Skip to content

Commit 7e4a83b

Browse files
committed
add scripts
1 parent 4ca1a19 commit 7e4a83b

File tree

3 files changed

+26
-11
lines changed

3 files changed

+26
-11
lines changed

.vscode/launch.json

+19
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
{
2+
// Use IntelliSense to learn about possible attributes.
3+
// Hover to view descriptions of existing attributes.
4+
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
5+
"version": "0.2.0",
6+
"configurations": [
7+
{
8+
"name": "Python: Current File",
9+
"type": "python",
10+
"request": "launch",
11+
"program": "${file}",
12+
"console": "integratedTerminal",
13+
"args": [
14+
"--root_path", "/media/nzioulis/New Volume/iti-688/Cleaned",
15+
"--trajectory_path", "/media/nzioulis/New Volume/iti-688/trajectories/",
16+
]
17+
}
18+
]
19+
}

.vscode/settings.json

+3
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
{
2+
"python.pythonPath": "/home/nzioulis/Documents/venv/deeplearning/bin/python3.6"
3+
}

visualize_dataset.py

+4-11
Original file line numberDiff line numberDiff line change
@@ -25,13 +25,6 @@ def parse_arguments(args):
2525
parser.add_argument("--frame_list",nargs="*", type=int, default = [0,1], help = "List of frames to be included")
2626
parser.add_argument("--types_list",nargs="*", type=str, default = ["colour", "depth","silhouette","normal"], help = "List of different modalities to be loaded")
2727

28-
# other
29-
parser.add_argument('-n','--name', type=str, default='default_name', help='The name of this train/test. Used when storing information.')
30-
31-
parser.add_argument("--visdom", type=str, nargs='?', default="localhost", const="127.0.0.1", help = "Visdom server IP (port defaults to 8097)")
32-
parser.add_argument("--visdom_iters", type=int, default=400, help = "Iteration interval that results will be reported at the visdom server for visualization.")
33-
parser.add_argument("--seed", type=int, default=1337, help="Fixed manual seed, zero means no seeding.")
34-
3528

3629
return parser.parse_known_args(args)
3730

@@ -56,10 +49,10 @@ def parse_arguments(args):
5649
#TODO: add a visualisation script to be added in the UAVA dataset
5750
for batch_id, batch in enumerate(train_set):
5851
#exocentric -- frame_0
59-
viz.show_images_grid(batch['exocentric'][0]["colour"], "all_colors")
60-
viz.show_depths_grid(batch['exocentric'][0]["depth"], "all_depths")
61-
viz.show_normals_grid(batch['exocentric'][0]["normal"], "all_normals")
62-
viz.show_images_grid(batch['exocentric'][0]["silhouette"], "silhouettes")
52+
viz.show_images_grid(batch['exocentric'][0]["colour"], "Exocentric view colour")
53+
viz.show_depths_grid(batch['exocentric'][0]["depth"], "Exocentric view depth")
54+
viz.show_normals_grid(batch['exocentric'][0]["normal"], "Exocentric view normals")
55+
viz.show_images_grid(batch['exocentric'][0]["silhouette"], "Exocentric view silhouette masks")
6356
#egocentric -- frame_0
6457
viz.show_images_grid(batch['egocentric'][0]["colour"], "Egocentric view colour")
6558
viz.show_depths_grid(batch['egocentric'][0]["depth"], "Egocentric view depth")

0 commit comments

Comments
 (0)