Subject | Hash | Author | Date (UTC) |
---|---|---|---|
prepare bike 20s q100 | 556a4d8f695ca154a5db42b363a276d7ab233434 | Thai Thien | 2020-12-15 16:46:08 |
VID_20201204_133931_404 | 8264dcb4d5389ed8ef2920eb3419cfc505b6a09c | Thai Thien | 2020-12-13 11:31:10 |
add VID_20201204_133931_404 | a36ba35ca7c54dcf552d17a08fa8c87720ce1c69 | Thai Thien | 2020-12-13 11:30:24 |
print total length on console | 6a956d93489926e99909d234bf4a7b13d39aad56 | Thai Thien | 2020-12-13 11:24:56 |
fix input of save_density_map | dc2f014fdd570909b811ac77629f0c8c0ca156da | Thai Thien | 2020-12-13 11:21:37 |
detach cpu for save_density_map | f41362501a0bc3578de0a48e6cbe97966a66677c | Thai Thien | 2020-12-13 11:19:45 |
predict_video_server pred.detach().cpu().numpy() | 3c23cd51890f8b791be91ad8eb74bdba9ace905a | Thai Thien | 2020-12-13 11:14:51 |
remove stuff | 212a4a703ae301fb90a0f292d8ccf19a1026e071 | Thai Thien | 2020-12-13 11:10:28 |
lot of code | 6b7a87a75171b0ec8ffe85960a38dd68e787e8f7 | Thai Thien | 2020-12-13 11:06:50 |
WIP | a6531891669668e489f67e90e862605ccb234911 | Thai Thien | 2020-12-12 16:02:50 |
video dataset | 59decbaabe7942bd19934050a57867ec485ba1f7 | Thai Thien | 2020-12-12 11:03:11 |
generate density map | f233f78f8c4fe17df088338d1b03e07f5a575896 | Thai Thien | 2020-12-12 03:44:15 |
ccnn_bike | 39a8223af58efc64f44f95245ade87f7f66c679a | Thai Thien | 2020-12-09 10:14:03 |
load_data_shanghaitech_non_overlap_test_with_densitygt dont need mat | 9bd1f7ce4931f3a172d8bb2858f74993eb8fd634 | Thai Thien | 2020-12-09 10:07:42 |
eval_adamw1_bigtail13i_t1_bike | e587492003ee8cc117eb4c41a2f66958e228ed70 | Thai Thien | 2020-12-09 10:04:39 |
. | f7f3c06081bcc2a46f438f58ca3302db047df7fb | Thai Thien | 2020-12-08 18:45:38 |
normal both ssim and psnr | 8c3dce50a264bea46fae09ec9390ebaa188134ed | Thai Thien | 2020-12-06 16:46:35 |
forgot the abs | da4b718d0e4c4e826fec189297807b478fa05bfc | Thai Thien | 2020-12-06 16:41:51 |
psnr abs 255 | ffaf76a25ca6c15ed8d36a77c21bd6d9278ab690 | Thai Thien | 2020-12-06 16:40:51 |
normalize 255 | d4e37756c87bb32afdd34ee368e0b5e826b9901f | Thai Thien | 2020-12-06 16:38:02 |
File | Lines added | Lines deleted |
---|---|---|
dataset_script/generate_frame_from_video.py | 24 | 0 |
dataset_script/my-bike-fixed.py | 2 | 2 |
dataset_script/mybike_train_validate_split.py | 2 | 2 |
File dataset_script/generate_frame_from_video.py added (mode: 100644) (index 0000000..cec44e9) | |||
1 | from torchvision.io.video import read_video, write_video | ||
2 | from torchvision.io.image import write_jpeg | ||
3 | import torch | ||
4 | from args_util import meow_parse | ||
5 | from data_flow import get_predict_video_dataloader | ||
6 | from models import create_model | ||
7 | import os | ||
8 | from visualize_util import save_density_map_normalize, save_density_map | ||
9 | |||
10 | VIDEO_PATH = "/home/tt/Videos/VID_20201204_133931_404.mp4" | ||
11 | OUTPUT_PATH = "/data/my_crowd_image/video_bike_20_q100" | ||
12 | v, a, info = read_video(VIDEO_PATH, pts_unit='sec') | ||
13 | print(v.shape) | ||
14 | length = v.shape[0] | ||
15 | print(length) | ||
16 | |||
17 | count = 0 | ||
18 | for i in range(length): | ||
19 | if (i% 20 == 0): | ||
20 | frame = v[i] | ||
21 | frame = frame.permute(2, 0, 1) | ||
22 | file_out_path = os.path.join(OUTPUT_PATH, "IMG_" + str(i) + ".jpg") | ||
23 | write_jpeg(frame, file_out_path, quality=100) | ||
24 | print(file_out_path) |
File dataset_script/my-bike-fixed.py changed (mode: 100644) (index a573976..48a6168) | |||
... | ... | def preprocess_args_parse(): | |
39 | 39 | return arg | return arg |
40 | 40 | ||
41 | 41 | # where it have /images | # where it have /images |
42 | __DATASET_ROOT = "/data/my_crowd_image/dataset_batch1245/mybikedata/unsort" | ||
42 | __DATASET_ROOT = "/data/my_crowd_image/data_bike_20_q100/unsort" | ||
43 | 43 | ||
44 | 44 | # where should it write ground-truth-h5 | # where should it write ground-truth-h5 |
45 | __OUTPUT_NAME = "/data/my_crowd_image/dataset_batch1245/mybikedata/unsort" | ||
45 | __OUTPUT_NAME = "/data/my_crowd_image/data_bike_20_q100/unsort" | ||
46 | 46 | ||
47 | 47 | # args = preprocess_args_parse() | # args = preprocess_args_parse() |
48 | 48 | # __DATASET_ROOT = args.root | # __DATASET_ROOT = args.root |
File dataset_script/mybike_train_validate_split.py changed (mode: 100644) (index 8533e10..f3681dd) | |||
... | ... | if __name__ == "__main__": | |
25 | 25 | ||
26 | 26 | # should contain 3 sub-folder: image, ground-truth, ground-truth-h5 | # should contain 3 sub-folder: image, ground-truth, ground-truth-h5 |
27 | 27 | # hey, no trailing slash | # hey, no trailing slash |
28 | DATA_PATH = "/data/my_crowd_image/dataset_batch1245/mybikedata/unsort" | ||
29 | OUTPUT_PREFIX = "/data/my_crowd_image/dataset_batch1245/mybikedata" | ||
28 | DATA_PATH = "/data/my_crowd_image/data_bike_20_q100/unsort" | ||
29 | OUTPUT_PREFIX = "/data/my_crowd_image/data_bike_20_q100/" | ||
30 | 30 | # args = train_test_split_parse() | # args = train_test_split_parse() |
31 | 31 | # DATA_PATH = args.input | # DATA_PATH = args.input |
32 | 32 |