File hard_code_variable.py changed (mode: 100644) (index b844ba1..b212186) |
|
1 |
|
import os |
|
2 |
|
import glob |
|
3 |
|
|
|
4 |
|
|
1 |
5 |
class HardCodeVariable(): |
class HardCodeVariable(): |
2 |
6 |
""" |
""" |
3 |
7 |
Where did you put your data ? |
Where did you put your data ? |
|
... |
... |
class HardCodeVariable(): |
5 |
9 |
""" |
""" |
6 |
10 |
def __init__(self): |
def __init__(self): |
7 |
11 |
self.UCF_CC_50_PATH = "/data/cv_data/UCFCrowdCountingDataset_CVPR13_with_people_density_map/UCF_CC_50" |
self.UCF_CC_50_PATH = "/data/cv_data/UCFCrowdCountingDataset_CVPR13_with_people_density_map/UCF_CC_50" |
8 |
|
self.SHANGHAITECH_PATH = "/data/ShanghaiTech/part_A/train_data" |
|
|
12 |
|
self.SHANGHAITECH_PATH = "/data/ShanghaiTech" |
|
13 |
|
self.SHANGHAITECH_PATH_PART_A = "/data/ShanghaiTech/part_A/" |
|
14 |
|
self.SHANGHAITECH_PATH_PART_B = "/data/ShanghaiTech/part_B/" |
|
15 |
|
self.SHANGHAITECH_PATH_TRAIN_POSTFIX = "train_data" |
|
16 |
|
self.SHANGHAITECH_PATH_TEST_POSTFIX = "test_data" |
|
17 |
|
|
|
18 |
|
|
|
19 |
|
if __name__ == "__main__": |
|
20 |
|
""" |
|
21 |
|
Test if path is working |
|
22 |
|
""" |
|
23 |
|
hard_code =HardCodeVariable() |
|
24 |
|
if os.path.exists(hard_code.SHANGHAITECH_PATH): |
|
25 |
|
print("exist " + hard_code.SHANGHAITECH_PATH) |
|
26 |
|
print("let see if we have train, test folder") |
|
27 |
|
train_path_a = os.path.join(hard_code.SHANGHAITECH_PATH_PART_A, hard_code.SHANGHAITECH_PATH_TRAIN_POSTFIX) |
|
28 |
|
train_path_b = os.path.join(hard_code.SHANGHAITECH_PATH_PART_B, hard_code.SHANGHAITECH_PATH_TRAIN_POSTFIX) |
|
29 |
|
test_path_a = os.path.join(hard_code.SHANGHAITECH_PATH_PART_A, hard_code.SHANGHAITECH_PATH_TEST_POSTFIX) |
|
30 |
|
test_path_b = os.path.join(hard_code.SHANGHAITECH_PATH_PART_B, hard_code.SHANGHAITECH_PATH_TEST_POSTFIX) |
|
31 |
|
if os.path.exists(train_path_a): |
|
32 |
|
print("exist " + train_path_a) |
|
33 |
|
if os.path.exists(train_path_b): |
|
34 |
|
print("exist " + train_path_b) |
|
35 |
|
if os.path.exists(test_path_a): |
|
36 |
|
print("exist " + test_path_a) |
|
37 |
|
if os.path.exists(test_path_b): |
|
38 |
|
print("exist " + test_path_b) |
|
39 |
|
print("count number of image") |
|
40 |
|
image_folder_list = [train_path_a, train_path_b, test_path_a, test_path_b] |
|
41 |
|
for image_root_path in image_folder_list: |
|
42 |
|
image_path_list = glob.glob(os.path.join(image_root_path, "images", "*.jpg")) |
|
43 |
|
count_img = len(image_path_list) |
|
44 |
|
first_img = image_path_list[0] |
|
45 |
|
print("in folder " + image_root_path) |
|
46 |
|
print("--- there are total " + str(count_img)) |
|
47 |
|
print('--- first img ' + first_img) |
File visualize_data_loader.py changed (mode: 100644) (index b6d376e..90d8de0) |
... |
... |
def visualize_shanghaitech_pacnn_with_perspective(): |
53 |
53 |
HARD_CODE = HardCodeVariable() |
HARD_CODE = HardCodeVariable() |
54 |
54 |
saved_folder = "visualize/test_dataloader" |
saved_folder = "visualize/test_dataloader" |
55 |
55 |
os.makedirs(saved_folder, exist_ok=True) |
os.makedirs(saved_folder, exist_ok=True) |
56 |
|
DATA_PATH = HARD_CODE.SHANGHAITECH_PATH |
|
|
56 |
|
DATA_PATH = HARD_CODE.SHANGHAITECH_PATH_PART_A |
57 |
57 |
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) |
58 |
58 |
test_list = None |
test_list = None |
59 |
59 |
|
|