Python 3.8.11 (default, Aug 3 2021, 15:09:35)
Type "copyright", "credits" or "license" for more information.
IPython 7.22.0 -- An enhanced Interactive Python.
Restarting kernel...
In [1]: runfile('/home/min/a/tibrayev/RESEARCH/active_fable/fable_v2_falcon/FALcon_collect_samples_for_WSOD.py', wdir='/home/min/a/tibrayev/RESEARCH/active_fable/fable_v2_falcon')
/home/min/a/tibrayev/miniconda3/envs/torch_1.9_torchvision_10.0/lib/python3.8/site-packages/tqdm/auto.py:22: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html
from .autonotebook import tqdm as notebook_tqdm
Selected VGG configuration (vgg11) was loaded from checkpoint: ./results/cub/wsol_method_PSOL/trained_on_trainval_split_evaluated_on_test_split/arch_vgg11_pretrained_init_normalization_none_seed_16/model.pth
FALcon (localization) model:
VGG(
(features): Sequential(
(0): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(1): ReLU(inplace=True)
(2): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)
(3): Conv2d(64, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(4): ReLU(inplace=True)
(5): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)
(6): Conv2d(128, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(7): ReLU(inplace=True)
(8): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(9): ReLU(inplace=True)
(10): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)
(11): Conv2d(256, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(12): ReLU(inplace=True)
(13): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(14): ReLU(inplace=True)
(15): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)
(16): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(17): ReLU(inplace=True)
(18): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(19): ReLU(inplace=True)
(20): AdaptiveAvgPool2d(output_size=(1, 1))
)
(fovea_control): Sequential(
(0): Linear(in_features=512, out_features=256, bias=True)
(1): ReLU(inplace=True)
(2): Linear(in_features=256, out_features=128, bias=True)
(3): ReLU(inplace=True)
(4): Linear(in_features=128, out_features=4, bias=True)
)
(saccade_control): Sequential(
(0): Linear(in_features=512, out_features=256, bias=True)
(1): ReLU(inplace=True)
(2): Linear(in_features=256, out_features=1, bias=True)
)
)
/home/min/a/tibrayev/miniconda3/envs/torch_1.9_torchvision_10.0/lib/python3.8/site-packages/torch/nn/functional.py:718: UserWarning: Named tensors and all their associated APIs are an experimental feature and subject to change. Please do not use them for anything important until they are released as stable. (Triggered internally at /opt/conda/conda-bld/pytorch_1623448278899/work/c10/core/TensorImpl.h:1156.)
return torch.max_pool2d(input, kernel_size, stride, padding, dilation, ceil_mode)
100/4952 requested samples processed!
200/4952 requested samples processed!
300/4952 requested samples processed!
400/4952 requested samples processed!
500/4952 requested samples processed!
600/4952 requested samples processed!
700/4952 requested samples processed!
800/4952 requested samples processed!
900/4952 requested samples processed!
1000/4952 requested samples processed!
1100/4952 requested samples processed!
1200/4952 requested samples processed!
1300/4952 requested samples processed!
1400/4952 requested samples processed!
1500/4952 requested samples processed!
1600/4952 requested samples processed!
1700/4952 requested samples processed!
1800/4952 requested samples processed!
1900/4952 requested samples processed!
2000/4952 requested samples processed!
2100/4952 requested samples processed!
2200/4952 requested samples processed!
2300/4952 requested samples processed!
2400/4952 requested samples processed!
2500/4952 requested samples processed!
2600/4952 requested samples processed!
2700/4952 requested samples processed!
2800/4952 requested samples processed!
2900/4952 requested samples processed!
3000/4952 requested samples processed!
3100/4952 requested samples processed!
3200/4952 requested samples processed!
3300/4952 requested samples processed!
3400/4952 requested samples processed!
3500/4952 requested samples processed!
3600/4952 requested samples processed!
3700/4952 requested samples processed!
3800/4952 requested samples processed!
3900/4952 requested samples processed!
4000/4952 requested samples processed!
4100/4952 requested samples processed!
4200/4952 requested samples processed!
4300/4952 requested samples processed!
4400/4952 requested samples processed!
4500/4952 requested samples processed!
4600/4952 requested samples processed!
4700/4952 requested samples processed!
4800/4952 requested samples processed!
4900/4952 requested samples processed!
4952/4952 requested samples processed!
In [2]: len(collected_samples)
Out[2]: 4952
In [3]:
...: from voc_evaluator import do_voc_evaluation
...:
...: results_voc = do_voc_evaluation(collected_samples)
In [4]: results_voc
Out[4]:
{'ap': array([ nan, 0. , 0. , 0.1256, 0. , 0. , 0. , 0. , 0. , 0. , 0. , 0. , 0. , 0. , 0. , 0. , 0. , 0. , 0. ,
0. , 0. ]),
'map': 0.00628124758839952}
In [5]: from voc_evaluator import calc_detection_voc_prec_rec
In [6]: prec, rec = calc_detection_voc_prec_rec(sample_collection = sample_collection, iou_thresh = 0.5)
Traceback (most recent call last):
File "<ipython-input-6-03236e799b2f>", line 1, in <module>
prec, rec = calc_detection_voc_prec_rec(sample_collection = sample_collection, iou_thresh = 0.5)
NameError: name 'sample_collection' is not defined
In [7]: prec, rec = calc_detection_voc_prec_rec(sample_collection = collected_samples, iou_thresh = 0.5)
In [8]: