Skip to content

Commit b49e4f0

Browse files
committed
train without augmentation
0 parents  commit b49e4f0

17 files changed

Lines changed: 687 additions & 0 deletions

.gitignore

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
__pycache__/*
2+
tmp/*
3+
utils/__pycache__/*
4+
Shanghai/*
5+
.gitattributes

ckpts/model.pth.tar

Whitespace-only changes.

data/metrics.json

Whitespace-only changes.

make_dataset.py

Lines changed: 52 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,52 @@
1+
import os
2+
import sys
3+
import logging
4+
import argparse
5+
6+
from utils import dataset
7+
8+
9+
def main(dataset_dir, subset):
10+
11+
if not os.path.exists(dataset_dir):
12+
raise OSError('Directory {} does not exist'.format(dataset))
13+
14+
if subset == 'Shanghai_A':
15+
for split in ['train_data', 'test_data']:
16+
section_dir = os.path.join(dataset_dir, 'part_A_final', split)
17+
save_dir = os.path.join(section_dir, 'densities')
18+
19+
# Save dir is where all density maps are going to be saved
20+
if not os.path.exists(save_dir):
21+
logging.info('Creating {} directory'.format(save_dir))
22+
os.mkdir(save_dir)
23+
data_loader = dataset.Shanghai_A(section_dir, save_dir)
24+
data_loader.create_groundtruth()
25+
26+
elif subset == 'Shanghai_B':
27+
for split in ['train_data', 'test_data']:
28+
section_dir = os.path.join(dataset_dir, 'part_B_final', split)
29+
save_dir = os.path.join(section_dir, 'densities')
30+
31+
# Save dir is where all density maps are going to be saved
32+
if not os.path.exists(save_dir):
33+
logging.info('Creating {} directory'.format(save_dir))
34+
os.mkdir(save_dir)
35+
36+
data_loader = dataset.Shanghai_B(section_dir, save_dir)
37+
data_loader.create_groundtruth()
38+
39+
else:
40+
raise ValueError('Subset not available for the moment')
41+
42+
43+
def parse_arguments(argv):
44+
parser = argparse.ArgumentParser()
45+
parser.add_argument('dataset_dir', type=str, help='Path to dataset directory')
46+
parser.add_argument('subset', type=str, help='Shanghai_A or Shanghai_B for the moment')
47+
return vars(parser.parse_args())
48+
49+
50+
if __name__ == '__main__':
51+
logging.basicConfig(level=logging.INFO)
52+
main(**parse_arguments(sys.argv[1:]))

part_A/test.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
["Shanghai/part_A_final/test_data/images/IMG_73.jpg", "Shanghai/part_A_final/test_data/images/IMG_38.jpg", "Shanghai/part_A_final/test_data/images/IMG_54.jpg", "Shanghai/part_A_final/test_data/images/IMG_145.jpg", "Shanghai/part_A_final/test_data/images/IMG_70.jpg", "Shanghai/part_A_final/test_data/images/IMG_91.jpg", "Shanghai/part_A_final/test_data/images/IMG_143.jpg", "Shanghai/part_A_final/test_data/images/IMG_173.jpg", "Shanghai/part_A_final/test_data/images/IMG_100.jpg", "Shanghai/part_A_final/test_data/images/IMG_3.jpg", "Shanghai/part_A_final/test_data/images/IMG_95.jpg", "Shanghai/part_A_final/test_data/images/IMG_104.jpg", "Shanghai/part_A_final/test_data/images/IMG_18.jpg", "Shanghai/part_A_final/test_data/images/IMG_181.jpg", "Shanghai/part_A_final/test_data/images/IMG_151.jpg", "Shanghai/part_A_final/test_data/images/IMG_169.jpg", "Shanghai/part_A_final/test_data/images/IMG_121.jpg", "Shanghai/part_A_final/test_data/images/IMG_14.jpg", "Shanghai/part_A_final/test_data/images/IMG_129.jpg", "Shanghai/part_A_final/test_data/images/IMG_174.jpg", "Shanghai/part_A_final/test_data/images/IMG_109.jpg", "Shanghai/part_A_final/test_data/images/IMG_172.jpg", "Shanghai/part_A_final/test_data/images/IMG_71.jpg", "Shanghai/part_A_final/test_data/images/IMG_74.jpg", "Shanghai/part_A_final/test_data/images/IMG_161.jpg", "Shanghai/part_A_final/test_data/images/IMG_43.jpg", "Shanghai/part_A_final/test_data/images/IMG_28.jpg", "Shanghai/part_A_final/test_data/images/IMG_67.jpg", "Shanghai/part_A_final/test_data/images/IMG_152.jpg", "Shanghai/part_A_final/test_data/images/IMG_130.jpg", "Shanghai/part_A_final/test_data/images/IMG_93.jpg", "Shanghai/part_A_final/test_data/images/IMG_156.jpg", "Shanghai/part_A_final/test_data/images/IMG_163.jpg", "Shanghai/part_A_final/test_data/images/IMG_12.jpg", "Shanghai/part_A_final/test_data/images/IMG_177.jpg", "Shanghai/part_A_final/test_data/images/IMG_46.jpg", "Shanghai/part_A_final/test_data/images/IMG_120.jpg", "Shanghai/part_A_final/test_data/images/IMG_24.jpg", "Shanghai/part_A_final/test_data/images/IMG_47.jpg", "Shanghai/part_A_final/test_data/images/IMG_9.jpg", "Shanghai/part_A_final/test_data/images/IMG_167.jpg", "Shanghai/part_A_final/test_data/images/IMG_141.jpg", "Shanghai/part_A_final/test_data/images/IMG_136.jpg", "Shanghai/part_A_final/test_data/images/IMG_15.jpg", "Shanghai/part_A_final/test_data/images/IMG_148.jpg", "Shanghai/part_A_final/test_data/images/IMG_112.jpg", "Shanghai/part_A_final/test_data/images/IMG_8.jpg", "Shanghai/part_A_final/test_data/images/IMG_164.jpg", "Shanghai/part_A_final/test_data/images/IMG_65.jpg", "Shanghai/part_A_final/test_data/images/IMG_44.jpg", "Shanghai/part_A_final/test_data/images/IMG_61.jpg", "Shanghai/part_A_final/test_data/images/IMG_179.jpg", "Shanghai/part_A_final/test_data/images/IMG_56.jpg", "Shanghai/part_A_final/test_data/images/IMG_131.jpg", "Shanghai/part_A_final/test_data/images/IMG_85.jpg", "Shanghai/part_A_final/test_data/images/IMG_139.jpg", "Shanghai/part_A_final/test_data/images/IMG_89.jpg", "Shanghai/part_A_final/test_data/images/IMG_26.jpg", "Shanghai/part_A_final/test_data/images/IMG_128.jpg", "Shanghai/part_A_final/test_data/images/IMG_32.jpg", "Shanghai/part_A_final/test_data/images/IMG_119.jpg", "Shanghai/part_A_final/test_data/images/IMG_60.jpg", "Shanghai/part_A_final/test_data/images/IMG_165.jpg", "Shanghai/part_A_final/test_data/images/IMG_166.jpg", "Shanghai/part_A_final/test_data/images/IMG_154.jpg", "Shanghai/part_A_final/test_data/images/IMG_64.jpg", "Shanghai/part_A_final/test_data/images/IMG_59.jpg", "Shanghai/part_A_final/test_data/images/IMG_78.jpg", "Shanghai/part_A_final/test_data/images/IMG_135.jpg", "Shanghai/part_A_final/test_data/images/IMG_114.jpg", "Shanghai/part_A_final/test_data/images/IMG_4.jpg", "Shanghai/part_A_final/test_data/images/IMG_175.jpg", "Shanghai/part_A_final/test_data/images/IMG_96.jpg", "Shanghai/part_A_final/test_data/images/IMG_58.jpg", "Shanghai/part_A_final/test_data/images/IMG_146.jpg", "Shanghai/part_A_final/test_data/images/IMG_92.jpg", "Shanghai/part_A_final/test_data/images/IMG_83.jpg", "Shanghai/part_A_final/test_data/images/IMG_105.jpg", "Shanghai/part_A_final/test_data/images/IMG_34.jpg", "Shanghai/part_A_final/test_data/images/IMG_111.jpg", "Shanghai/part_A_final/test_data/images/IMG_101.jpg", "Shanghai/part_A_final/test_data/images/IMG_17.jpg", "Shanghai/part_A_final/test_data/images/IMG_22.jpg", "Shanghai/part_A_final/test_data/images/IMG_107.jpg", "Shanghai/part_A_final/test_data/images/IMG_103.jpg", "Shanghai/part_A_final/test_data/images/IMG_127.jpg", "Shanghai/part_A_final/test_data/images/IMG_106.jpg", "Shanghai/part_A_final/test_data/images/IMG_33.jpg", "Shanghai/part_A_final/test_data/images/IMG_125.jpg", "Shanghai/part_A_final/test_data/images/IMG_144.jpg", "Shanghai/part_A_final/test_data/images/IMG_138.jpg", "Shanghai/part_A_final/test_data/images/IMG_29.jpg", "Shanghai/part_A_final/test_data/images/IMG_155.jpg", "Shanghai/part_A_final/test_data/images/IMG_115.jpg", "Shanghai/part_A_final/test_data/images/IMG_149.jpg", "Shanghai/part_A_final/test_data/images/IMG_72.jpg", "Shanghai/part_A_final/test_data/images/IMG_134.jpg", "Shanghai/part_A_final/test_data/images/IMG_153.jpg", "Shanghai/part_A_final/test_data/images/IMG_133.jpg", "Shanghai/part_A_final/test_data/images/IMG_116.jpg", "Shanghai/part_A_final/test_data/images/IMG_51.jpg", "Shanghai/part_A_final/test_data/images/IMG_123.jpg", "Shanghai/part_A_final/test_data/images/IMG_41.jpg", "Shanghai/part_A_final/test_data/images/IMG_81.jpg", "Shanghai/part_A_final/test_data/images/IMG_55.jpg", "Shanghai/part_A_final/test_data/images/IMG_19.jpg", "Shanghai/part_A_final/test_data/images/IMG_48.jpg", "Shanghai/part_A_final/test_data/images/IMG_7.jpg", "Shanghai/part_A_final/test_data/images/IMG_158.jpg", "Shanghai/part_A_final/test_data/images/IMG_80.jpg", "Shanghai/part_A_final/test_data/images/IMG_118.jpg", "Shanghai/part_A_final/test_data/images/IMG_84.jpg", "Shanghai/part_A_final/test_data/images/IMG_76.jpg", "Shanghai/part_A_final/test_data/images/IMG_90.jpg", "Shanghai/part_A_final/test_data/images/IMG_30.jpg", "Shanghai/part_A_final/test_data/images/IMG_53.jpg", "Shanghai/part_A_final/test_data/images/IMG_176.jpg", "Shanghai/part_A_final/test_data/images/IMG_68.jpg", "Shanghai/part_A_final/test_data/images/IMG_168.jpg", "Shanghai/part_A_final/test_data/images/IMG_63.jpg", "Shanghai/part_A_final/test_data/images/IMG_86.jpg", "Shanghai/part_A_final/test_data/images/IMG_20.jpg", "Shanghai/part_A_final/test_data/images/IMG_147.jpg", "Shanghai/part_A_final/test_data/images/IMG_108.jpg", "Shanghai/part_A_final/test_data/images/IMG_45.jpg", "Shanghai/part_A_final/test_data/images/IMG_82.jpg", "Shanghai/part_A_final/test_data/images/IMG_10.jpg", "Shanghai/part_A_final/test_data/images/IMG_49.jpg", "Shanghai/part_A_final/test_data/images/IMG_5.jpg", "Shanghai/part_A_final/test_data/images/IMG_87.jpg", "Shanghai/part_A_final/test_data/images/IMG_160.jpg", "Shanghai/part_A_final/test_data/images/IMG_170.jpg", "Shanghai/part_A_final/test_data/images/IMG_178.jpg", "Shanghai/part_A_final/test_data/images/IMG_126.jpg", "Shanghai/part_A_final/test_data/images/IMG_162.jpg", "Shanghai/part_A_final/test_data/images/IMG_23.jpg", "Shanghai/part_A_final/test_data/images/IMG_2.jpg", "Shanghai/part_A_final/test_data/images/IMG_1.jpg", "Shanghai/part_A_final/test_data/images/IMG_171.jpg", "Shanghai/part_A_final/test_data/images/IMG_52.jpg", "Shanghai/part_A_final/test_data/images/IMG_42.jpg", "Shanghai/part_A_final/test_data/images/IMG_21.jpg", "Shanghai/part_A_final/test_data/images/IMG_132.jpg", "Shanghai/part_A_final/test_data/images/IMG_31.jpg", "Shanghai/part_A_final/test_data/images/IMG_6.jpg", "Shanghai/part_A_final/test_data/images/IMG_140.jpg", "Shanghai/part_A_final/test_data/images/IMG_40.jpg", "Shanghai/part_A_final/test_data/images/IMG_142.jpg", "Shanghai/part_A_final/test_data/images/IMG_36.jpg", "Shanghai/part_A_final/test_data/images/IMG_110.jpg", "Shanghai/part_A_final/test_data/images/IMG_77.jpg", "Shanghai/part_A_final/test_data/images/IMG_27.jpg", "Shanghai/part_A_final/test_data/images/IMG_137.jpg", "Shanghai/part_A_final/test_data/images/IMG_57.jpg", "Shanghai/part_A_final/test_data/images/IMG_159.jpg", "Shanghai/part_A_final/test_data/images/IMG_122.jpg", "Shanghai/part_A_final/test_data/images/IMG_113.jpg", "Shanghai/part_A_final/test_data/images/IMG_62.jpg", "Shanghai/part_A_final/test_data/images/IMG_124.jpg", "Shanghai/part_A_final/test_data/images/IMG_150.jpg", "Shanghai/part_A_final/test_data/images/IMG_39.jpg", "Shanghai/part_A_final/test_data/images/IMG_99.jpg", "Shanghai/part_A_final/test_data/images/IMG_37.jpg", "Shanghai/part_A_final/test_data/images/IMG_75.jpg", "Shanghai/part_A_final/test_data/images/IMG_157.jpg", "Shanghai/part_A_final/test_data/images/IMG_98.jpg", "Shanghai/part_A_final/test_data/images/IMG_66.jpg", "Shanghai/part_A_final/test_data/images/IMG_94.jpg", "Shanghai/part_A_final/test_data/images/IMG_13.jpg", "Shanghai/part_A_final/test_data/images/IMG_102.jpg", "Shanghai/part_A_final/test_data/images/IMG_180.jpg", "Shanghai/part_A_final/test_data/images/IMG_11.jpg", "Shanghai/part_A_final/test_data/images/IMG_25.jpg", "Shanghai/part_A_final/test_data/images/IMG_97.jpg", "Shanghai/part_A_final/test_data/images/IMG_182.jpg", "Shanghai/part_A_final/test_data/images/IMG_35.jpg", "Shanghai/part_A_final/test_data/images/IMG_50.jpg", "Shanghai/part_A_final/test_data/images/IMG_88.jpg", "Shanghai/part_A_final/test_data/images/IMG_79.jpg", "Shanghai/part_A_final/test_data/images/IMG_69.jpg", "Shanghai/part_A_final/test_data/images/IMG_117.jpg", "Shanghai/part_A_final/test_data/images/IMG_16.jpg"]

part_A/train.json

Lines changed: 1 addition & 0 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)