Skip to content

GitLab

Commit bac8c938 authored by BobYeah's avatar BobYeah
Browse files

update

parent 5c75685b
Hide whitespace changes
Inline Side-by-side
main.py
View file @ bac8c938
...@@ -25,9 +25,9 @@ IM_W = 640 ...@@ -25,9 +25,9 @@ IM_W = 640
N = 9 # number of input light field stack N = 9 # number of input light field stack
M = 2 # number of display layers M = 2 # number of display layers
DATA_FILE = "/home/yejiannan/Project/LightField/data/try" DATA_FILE = "/home/yejiannan/Project/deeplightfield/data/try"
DATA_JSON = "/home/yejiannan/Project/LightField/data/data.json" DATA_JSON = "/home/yejiannan/Project/deeplightfield/data/data.json"
OUTPUT_DIR = "/home/yejiannan/Project/LightField/output" OUTPUT_DIR = "/home/yejiannan/Project/deeplightfield/output"
class lightFieldDataLoader(torch.utils.data.dataset.Dataset): class lightFieldDataLoader(torch.utils.data.dataset.Dataset):
def __init__(self, file_dir_path, file_json, transforms=None): def __init__(self, file_dir_path, file_json, transforms=None):
......
Markdown is supported
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