Subject | Hash | Author | Date (UTC) |
---|---|---|---|
fix load model stuff | 8c5c9c985fb747dc6392c86aa5f49282f8d12297 | Thai Thien | 2020-12-18 15:44:21 |
import comet first | dd4426221a95d0916c8cc7c02c7587ada1d910b5 | Thai Thien | 2020-12-18 15:41:59 |
predict on frame from video | ec39a97372a15ad75811fdb3417d2e48078a30b2 | Thai Thien | 2020-12-18 15:40:36 |
fix normalize in data loader | 56a64c14dc62c96aba532d233ff2782414738d05 | Thai Thien | 2020-12-17 15:36:22 |
change gpu 5 | 46d32b3fc05909b61f56389b01c2e50e52937d9b | Thai Thien | 2020-12-16 16:40:47 |
eval_dccnn_bike_video_VID_20201204_133931_404_model_adamw1_bigtail13i_t4_bike20 | 95bb636a0b544b7a72fc4c65acb47472da468766 | Thai Thien | 2020-12-16 16:35:14 |
eval_dccnn_bike_video_VID_20201204_133931_404_model_adamw1_bigtail13i_t3_bike20 | 27740a0e21ca42490f5cfe516b2453f7be3a2483 | Thai Thien | 2020-12-16 16:17:12 |
batch 2 | 85433ffd51070572d7e743a2b4ef53f340b658d4 | Thai Thien | 2020-12-15 16:57:56 |
reduce batch size 3 | d2f145009c06f5e6e6e1e0270878ae275b611c2f | Thai Thien | 2020-12-15 16:55:46 |
t3 and t4 | 70e09414713ea22800d8e87a0dd984f18406c375 | Thai Thien | 2020-12-15 16:50:32 |
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 |
File | Lines added | Lines deleted |
---|---|---|
predict_image.py | 3 | 2 |
File predict_image.py changed (mode: 100644) (index bcaae93..395955c) | |||
... | ... | import torch | |
4 | 4 | from data_flow import get_predict_dataloader | from data_flow import get_predict_dataloader |
5 | 5 | from models.dccnn import DCCNN | from models.dccnn import DCCNN |
6 | 6 | from models.compact_cnn import CompactCNNV7 | from models.compact_cnn import CompactCNNV7 |
7 | from models import create_model | ||
7 | 8 | from visualize_util import save_density_map_normalize, save_density_map | from visualize_util import save_density_map_normalize, save_density_map |
8 | 9 | from args_util import meow_parse | from args_util import meow_parse |
9 | 10 | ||
... | ... | if __name__ == "__main__": | |
40 | 41 | NAME = args.task_id | NAME = args.task_id |
41 | 42 | INPUT_FOLDER = args.input | INPUT_FOLDER = args.input |
42 | 43 | OUTPUT_FOLDER = args.output | OUTPUT_FOLDER = args.output |
43 | MODEL = args.model | ||
44 | MODEL = args.load_model | ||
44 | 45 | input_list = [os.path.join(INPUT_FOLDER, dir) for dir in os.listdir(INPUT_FOLDER)] | input_list = [os.path.join(INPUT_FOLDER, dir) for dir in os.listdir(INPUT_FOLDER)] |
45 | 46 | loader = get_predict_dataloader(input_list) | loader = get_predict_dataloader(input_list) |
46 | 47 | loaded_file = torch.load(MODEL) | loaded_file = torch.load(MODEL) |
47 | model = CompactCNNV7() | ||
48 | model = create_model(model_name) | ||
48 | 49 | model.load_state_dict(loaded_file['model']) | model.load_state_dict(loaded_file['model']) |
49 | 50 | model.eval() | model.eval() |
50 | 51 | model = model.to(device) | model = model.to(device) |