File debug/verify_sha.py added (mode: 100644) (index 0000000..f91f783) |
|
1 |
|
import torch |
|
2 |
|
from models.meow_experiment.ccnn_tail import BigTail11i, BigTail10i, BigTail12i, BigTail13i, BigTail14i, BigTail15i |
|
3 |
|
from hard_code_variable import HardCodeVariable |
|
4 |
|
from data_util import ShanghaiTechDataPath |
|
5 |
|
from visualize_util import save_img, save_density_map |
|
6 |
|
import os |
|
7 |
|
import numpy as np |
|
8 |
|
from data_flow import get_train_val_list, get_dataloader, create_training_image_list |
|
9 |
|
import cv2 |
|
10 |
|
|
|
11 |
|
def visualize_evaluation_shanghaitech_keepfull(path=None): |
|
12 |
|
HARD_CODE = HardCodeVariable() |
|
13 |
|
if path==None: |
|
14 |
|
shanghaitech_data = ShanghaiTechDataPath(root= HARD_CODE.SHANGHAITECH_PATH) |
|
15 |
|
shanghaitech_data_part_a_train = shanghaitech_data.get_a().get_train().get() |
|
16 |
|
path = shanghaitech_data_part_a_train |
|
17 |
|
saved_folder = "visualize/verify_dataloader_shanghaitech" |
|
18 |
|
os.makedirs(saved_folder, exist_ok=True) |
|
19 |
|
train_list, val_list = get_train_val_list(path, test_size=0.2) |
|
20 |
|
test_list = None |
|
21 |
|
train_loader, val_loader, test_loader = get_dataloader(train_list, val_list, test_list, dataset_name="shanghaitech_keepfull", visualize_mode=True, |
|
22 |
|
debug=True) |
|
23 |
|
|
|
24 |
|
# do with train loader |
|
25 |
|
train_loader_iter = iter(train_loader) |
|
26 |
|
for i in range(len(train_loader)): |
|
27 |
|
img, label, count = next(train_loader_iter) |
|
28 |
|
save_img(img, os.path.join(saved_folder, "train_img_" + str(i) +".png")) |
|
29 |
|
save_path = os.path.join(saved_folder, "train_label_" + str(i) +".png") |
|
30 |
|
save_density_map(label.numpy()[0][0], save_path) |
|
31 |
|
|
|
32 |
|
|
|
33 |
|
visualize_evaluation_shanghaitech_keepfull() |
|
34 |
|
|
File train_script/learnstuff/l1/eval_adamw1_bigtail13i_t1_shb_on_sha.sh copied from file train_script/learnstuff/l1/adamw1_bigtail13i_t1_shb.sh (similarity 64%) (mode: 100644) (index cd3f8e4..b7d9f51) |
1 |
|
task="adamw1_bigtail13i_t1_shb" |
|
|
1 |
|
task="eval_adamw1_bigtail13i_t1_shb_on_sha" |
2 |
2 |
|
|
3 |
3 |
CUDA_VISIBLE_DEVICES=3 OMP_NUM_THREADS=2 PYTHONWARNINGS="ignore" HTTPS_PROXY="http://10.60.28.99:86" nohup python experiment_main.py \ |
CUDA_VISIBLE_DEVICES=3 OMP_NUM_THREADS=2 PYTHONWARNINGS="ignore" HTTPS_PROXY="http://10.60.28.99:86" nohup python experiment_main.py \ |
4 |
4 |
--task_id $task \ |
--task_id $task \ |
5 |
5 |
--note "adamW with extrem high lr and decay, msel1mean" \ |
--note "adamW with extrem high lr and decay, msel1mean" \ |
6 |
6 |
--model "BigTail13i" \ |
--model "BigTail13i" \ |
7 |
|
--input /data/rnd/thient/thient_data/shanghaitech_with_people_density_map/ShanghaiTech_3/part_B \ |
|
|
7 |
|
--input /data/rnd/thient/thient_data/shanghaitech_with_people_density_map/ShanghaiTech_3/part_A \ |
8 |
8 |
--lr 1e-3 \ |
--lr 1e-3 \ |
9 |
9 |
--decay 0.1 \ |
--decay 0.1 \ |
10 |
10 |
--loss_fn "MSEL1Mean" \ |
--loss_fn "MSEL1Mean" \ |
11 |
11 |
--batch_size 5 \ |
--batch_size 5 \ |
12 |
|
--datasetname shanghaitech_non_overlap \ |
|
|
12 |
|
--datasetname shanghaitech_keepfull \ |
13 |
13 |
--optim adamw \ |
--optim adamw \ |
|
14 |
|
--eval_only \ |
|
15 |
|
--load_model "saved_model_best/adamw1_bigtail13i_t1_shb/adamw1_bigtail13i_t1_shb_checkpoint_valid_mae=-7.574910521507263.pth" \ |
14 |
16 |
--cache \ |
--cache \ |
15 |
17 |
--epochs 1201 > logs/$task.log & |
--epochs 1201 > logs/$task.log & |
16 |
18 |
|
|