File data_flow.py changed (mode: 100644) (index baf3e25..906abd1) |
... |
... |
def data_augmentation(img, target): |
130 |
130 |
|
|
131 |
131 |
class ListDataset(Dataset): |
class ListDataset(Dataset): |
132 |
132 |
def __init__(self, root, shape=None, shuffle=True, transform=None, train=False, seen=0, batch_size=1, |
def __init__(self, root, shape=None, shuffle=True, transform=None, train=False, seen=0, batch_size=1, |
|
133 |
|
debug=False, |
133 |
134 |
num_workers=4, dataset_name="shanghaitech"): |
num_workers=4, dataset_name="shanghaitech"): |
134 |
135 |
""" |
""" |
135 |
136 |
if you have different image size, then batch_size must be 1 |
if you have different image size, then batch_size must be 1 |
|
... |
... |
class ListDataset(Dataset): |
138 |
139 |
:param shuffle: |
:param shuffle: |
139 |
140 |
:param transform: |
:param transform: |
140 |
141 |
:param train: |
:param train: |
|
142 |
|
:param debug: will print path of image |
141 |
143 |
:param seen: |
:param seen: |
142 |
144 |
:param batch_size: |
:param batch_size: |
143 |
145 |
:param num_workers: |
:param num_workers: |
|
... |
... |
class ListDataset(Dataset): |
151 |
153 |
self.lines = root |
self.lines = root |
152 |
154 |
self.transform = transform |
self.transform = transform |
153 |
155 |
self.train = train |
self.train = train |
|
156 |
|
self.debug = debug |
154 |
157 |
self.shape = shape |
self.shape = shape |
155 |
158 |
self.seen = seen |
self.seen = seen |
156 |
159 |
self.batch_size = batch_size |
self.batch_size = batch_size |
|
... |
... |
class ListDataset(Dataset): |
170 |
173 |
def __getitem__(self, index): |
def __getitem__(self, index): |
171 |
174 |
assert index <= len(self), 'index range error' |
assert index <= len(self), 'index range error' |
172 |
175 |
img_path = self.lines[index] |
img_path = self.lines[index] |
|
176 |
|
if self.debug: |
|
177 |
|
print(img_path) |
173 |
178 |
img, target = self.load_data_fn(img_path, self.train) |
img, target = self.load_data_fn(img_path, self.train) |
174 |
179 |
if self.transform is not None: |
if self.transform is not None: |
175 |
180 |
img = self.transform(img) |
img = self.transform(img) |
File visualize_data_loader.py changed (mode: 100644) (index ad2acdc..6307e90) |
... |
... |
import pytorch_ssim |
16 |
16 |
from hard_code_variable import HardCodeVariable |
from hard_code_variable import HardCodeVariable |
17 |
17 |
from visualize_util import save_img, save_density_map |
from visualize_util import save_img, save_density_map |
18 |
18 |
|
|
19 |
|
|
|
20 |
|
if __name__ == "__main__": |
|
|
19 |
|
def visualize_ucf_cc_50_pacnn(): |
21 |
20 |
HARD_CODE = HardCodeVariable() |
HARD_CODE = HardCodeVariable() |
|
21 |
|
saved_folder = "visualize/test_dataloader" |
|
22 |
|
os.makedirs(saved_folder, exist_ok=True) |
22 |
23 |
DATA_PATH = HARD_CODE.UCF_CC_50_PATH |
DATA_PATH = HARD_CODE.UCF_CC_50_PATH |
23 |
24 |
train_list, val_list = get_train_val_list(DATA_PATH, test_size=0.2) |
train_list, val_list = get_train_val_list(DATA_PATH, test_size=0.2) |
24 |
25 |
test_list = None |
test_list = None |
|
... |
... |
if __name__ == "__main__": |
33 |
34 |
]), |
]), |
34 |
35 |
train=True, |
train=True, |
35 |
36 |
batch_size=1, |
batch_size=1, |
36 |
|
num_workers=4, dataset_name="ucf_cc_50_pacnn"), |
|
|
37 |
|
num_workers=4, dataset_name="ucf_cc_50_pacnn", debug=True), |
37 |
38 |
batch_size=1, num_workers=4) |
batch_size=1, num_workers=4) |
38 |
39 |
|
|
39 |
40 |
img, label = next(iter(train_loader_pacnn)) |
img, label = next(iter(train_loader_pacnn)) |
40 |
41 |
|
|
41 |
42 |
print(img.shape) |
print(img.shape) |
42 |
|
save_img(img, "pacnn_loader_img.png") |
|
43 |
|
save_density_map(label[0].numpy()[0], "pacnn_loader_density1.png") |
|
44 |
|
save_density_map(label[1].numpy()[0], "pacnn_loader_density2.png") |
|
45 |
|
save_density_map(label[2].numpy()[0], "pacnn_loader_density3.png") |
|
|
43 |
|
save_img(img, os.path.join(saved_folder, "pacnn_loader_img.png")) |
|
44 |
|
save_density_map(label[0].numpy()[0], os.path.join(saved_folder,"pacnn_loader_density1.png")) |
|
45 |
|
save_density_map(label[1].numpy()[0], os.path.join(saved_folder,"pacnn_loader_density2.png")) |
|
46 |
|
save_density_map(label[2].numpy()[0], os.path.join(saved_folder,"pacnn_loader_density3.png")) |
|
47 |
|
|
|
48 |
|
|
46 |
49 |
|
|
|
50 |
|
|
|
51 |
|
|
|
52 |
|
if __name__ == "__main__": |
|
53 |
|
visualize_ucf_cc_50_pacnn() |