search

waxnkw / IETrans-SGG.pytorch

This is the code of ECCV 2022 (Oral) paper "Fine-Grained Scene Graph Generation with Data Transfer".
Other
89 stars 6 forks source link

Request for Alternative Download Link for Pre-trained Weights #16

Closed park-youngjae closed 1 year ago

park-youngjae commented 1 year ago

Thank you for your wonderful code and research.

However, I am unable to download your pre-trained weights with your download link. (https://thunlp.oss-cn-qingdao.aliyuncs.com/ietrans/vg1800_motif_sgcls_I90_E100.pth)

Can you please provide me with an alternative URL to download your weights?

For your information, I am located outside mainland China.

Thank you!

waxnkw commented 1 year ago

Please try this one https://drive.google.com/file/d/1eYXZmpvUhFc4-Jud2dEPDrF6NaDsTrkP/view?usp=sharing

park-youngjae commented 1 year ago

Thank you for your quick response. However, I got another issue dealing with your 'vg50-sgcls-I0.9E1.0-model_0030000.pth' weight file. I put weight file at 'visualization/demo_ims/' and changed name of file as 'last_checkpoint.pth'. However, I am getting below logs and below. Can you also explain exact directory where should I put your vg50-sgcls weight?

log---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- PyTorch version: 1.6.0 Is debug build: No CUDA used to build PyTorch: 10.1

OS: Ubuntu 18.04.3 LTS GCC version: (Ubuntu 7.4.0-1ubuntu1~18.04.1) 7.4.0 CMake version: Could not collect

Python version: 3.7 Is CUDA available: Yes CUDA runtime version: 10.1.243 GPU models and configuration: GPU 0: GeForce RTX 2080 Ti GPU 1: GeForce RTX 2080 Ti GPU 2: GeForce RTX 2080 Ti GPU 3: GeForce RTX 2080 Ti GPU 4: GeForce RTX 2080 Ti GPU 5: GeForce RTX 2080 Ti GPU 6: GeForce RTX 2080 Ti GPU 7: GeForce RTX 2080 Ti

Nvidia driver version: 418.87.00 cuDNN version: /usr/lib/x86_64-linux-gnu/libcudnn.so.7.6.5

Versions of relevant libraries: [pip] numpy==1.18.5 [pip] torch==1.6.0 [pip] torchvision==0.7.0 [conda] blas 1.0 mkl [conda] cudatoolkit 10.1.243 h6bb024c_0 [conda] mkl 2020.1 217 [conda] mkl-service 2.3.0 py37he904b0f_0 [conda] mkl_fft 1.1.0 py37h23d657b_0 [conda] mkl_random 1.1.1 py37h0573a6f_0 [conda] numpy 1.18.5 py37ha1c710e_0 [conda] numpy-base 1.18.5 py37hde5b4d6_0 [conda] pytorch 1.6.0 py3.7_cuda10.1.243_cudnn7.6.3_0 pytorch [conda] torchvision 0.7.0 py37_cu101 pytorch Pillow (7.2.0) 2023-04-02 08:13:22,877 maskrcnn_benchmark.data.build INFO: ---------------------------------------------------------------------------------------------------- 2023-04-02 08:13:22,878 maskrcnn_benchmark.data.build INFO: get dataset statistics... 2023-04-02 08:13:22,880 maskrcnn_benchmark.data.build INFO: Loading data statistics from: ./visualization/demo/50VG_stanford_filtered_with_attribute_train_statistics.cache 2023-04-02 08:13:22,880 maskrcnn_benchmark.data.build INFO: ---------------------------------------------------------------------------------------------------- loading word vectors from /glove/glove.6B.200d.pt

2023-04-02 08:13:57,560 maskrcnn_benchmark.utils.checkpoint INFO: Loading checkpoint from ./visualization/demo/last_checkpoint.pth 2023-04-02 08:13:59,233 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.bbox_embed.0.bias of shape (32,) 2023-04-02 08:13:59,233 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.bbox_embed.0.weight of shape (32, 9) 2023-04-02 08:13:59,233 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.bbox_embed.3.bias of shape (128,) 2023-04-02 08:13:59,233 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.bbox_embed.3.weight of shape (128, 32) 2023-04-02 08:13:59,233 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.pos_ffn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,233 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.pos_ffn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,233 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.pos_ffn.w_1.bias of shape (2048,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.pos_ffn.w_1.weight of shape (2048, 512, 1) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.pos_ffn.w_2.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.pos_ffn.w_2.weight of shape (512, 2048, 1) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.slf_attn.fc.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.slf_attn.fc.weight of shape (512, 512) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.slf_attn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.slf_attn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.slf_attn.w_ks.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.slf_attn.w_ks.weight of shape (512, 512) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.slf_attn.w_qs.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.slf_attn.w_qs.weight of shape (512, 512) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.slf_attn.w_vs.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.0.slf_attn.w_vs.weight of shape (512, 512) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.pos_ffn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.pos_ffn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.pos_ffn.w_1.bias of shape (2048,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.pos_ffn.w_1.weight of shape (2048, 512, 1) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.pos_ffn.w_2.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.pos_ffn.w_2.weight of shape (512, 2048, 1) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.slf_attn.fc.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.slf_attn.fc.weight of shape (512, 512) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.slf_attn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.slf_attn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.slf_attn.w_ks.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.slf_attn.w_ks.weight of shape (512, 512) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.slf_attn.w_qs.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.slf_attn.w_qs.weight of shape (512, 512) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.slf_attn.w_vs.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_edge.layer_stack.1.slf_attn.w_vs.weight of shape (512, 512) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.pos_ffn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,234 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.pos_ffn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.pos_ffn.w_1.bias of shape (2048,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.pos_ffn.w_1.weight of shape (2048, 512, 1) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.pos_ffn.w_2.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.pos_ffn.w_2.weight of shape (512, 2048, 1) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.slf_attn.fc.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.slf_attn.fc.weight of shape (512, 512) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.slf_attn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.slf_attn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.slf_attn.w_ks.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.slf_attn.w_ks.weight of shape (512, 512) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.slf_attn.w_qs.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.slf_attn.w_qs.weight of shape (512, 512) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.slf_attn.w_vs.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.0.slf_attn.w_vs.weight of shape (512, 512) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.pos_ffn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.pos_ffn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.pos_ffn.w_1.bias of shape (2048,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.pos_ffn.w_1.weight of shape (2048, 512, 1) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.pos_ffn.w_2.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.pos_ffn.w_2.weight of shape (512, 2048, 1) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.slf_attn.fc.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.slf_attn.fc.weight of shape (512, 512) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.slf_attn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.slf_attn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.slf_attn.w_ks.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.slf_attn.w_ks.weight of shape (512, 512) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.slf_attn.w_qs.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.slf_attn.w_qs.weight of shape (512, 512) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.slf_attn.w_vs.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.1.slf_attn.w_vs.weight of shape (512, 512) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.pos_ffn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,235 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.pos_ffn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.pos_ffn.w_1.bias of shape (2048,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.pos_ffn.w_1.weight of shape (2048, 512, 1) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.pos_ffn.w_2.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.pos_ffn.w_2.weight of shape (512, 2048, 1) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.slf_attn.fc.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.slf_attn.fc.weight of shape (512, 512) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.slf_attn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.slf_attn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.slf_attn.w_ks.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.slf_attn.w_ks.weight of shape (512, 512) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.slf_attn.w_qs.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.slf_attn.w_qs.weight of shape (512, 512) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.slf_attn.w_vs.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.2.slf_attn.w_vs.weight of shape (512, 512) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.pos_ffn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.pos_ffn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.pos_ffn.w_1.bias of shape (2048,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.pos_ffn.w_1.weight of shape (2048, 512, 1) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.pos_ffn.w_2.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.pos_ffn.w_2.weight of shape (512, 2048, 1) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.slf_attn.fc.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.slf_attn.fc.weight of shape (512, 512) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.slf_attn.layer_norm.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.slf_attn.layer_norm.weight of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.slf_attn.w_ks.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.slf_attn.w_ks.weight of shape (512, 512) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.slf_attn.w_qs.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.slf_attn.w_qs.weight of shape (512, 512) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.slf_attn.w_vs.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.context_obj.layer_stack.3.slf_attn.w_vs.weight of shape (512, 512) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.lin_edge.bias of shape (512,) 2023-04-02 08:13:59,236 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.lin_edge.weight of shape (512, 4808) 2023-04-02 08:13:59,237 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.lin_obj.bias of shape (512,) 2023-04-02 08:13:59,237 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.lin_obj.weight of shape (512, 4424) 2023-04-02 08:13:59,237 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.out_obj.bias of shape (151,) 2023-04-02 08:13:59,237 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.context_layer.out_obj.weight of shape (151, 512) 2023-04-02 08:13:59,237 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.ctx_compress.bias of shape (51,) 2023-04-02 08:13:59,237 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.ctx_compress.weight of shape (51, 1024) 2023-04-02 08:13:59,237 maskrcnn_benchmark.utils.model_serialization INFO: NO-MATCHING of current module: roi_heads.relation.predictor.freq_bias.obj_baseline.weight of shape (22801, 51) =====> ./visualization/demo/custom_data_info.json SAVED ! 2023-04-02 08:13:59,913 maskrcnn_benchmark.inference INFO: Start evaluation on 50VG_stanford_filtered_with_attribute_test dataset(3 images). 0%| | 0/3 [00:00<?, ?it/s]/workspace/Disaster/Code/IETrans/IETrans/maskrcnn_benchmark/structures/boxlist_ops.py:47: UserWarning: This overload of nonzero is deprecated: nonzero() Consider using one of the following signatures instead: nonzero(*, bool as_tuple) (Triggered internally at /opt/conda/conda-bld/pytorch_1595629403081/work/torch/csrc/utils/python_arg_parser.cpp:766.) (ws >= min_size) & (hs >= min_size) 100%|█████████████████████████████████████████████████████████████████████████████████████████| 3/3 [00:06<00:00, 2.22s/it] 2023-04-02 08:14:09,512 maskrcnn_benchmark.inference INFO: Total run time: 0:00:09.598739 (3.1995797952016196 s / img per device, on 1 devices) 2023-04-02 08:14:09,513 maskrcnn_benchmark.inference INFO: Model inference time: 0:00:06.131688 (2.043895880381266 s / img per device, on 1 devices) =====> ./visualization/demo/custom_prediction.pk SAVED !

Visualization Result ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- 4-windshield, along, 1-bag tensor(0.9997) 4-windshield, along, 0-windshield tensor(0.9995) 5-windshield, along, 4-windshield tensor(0.9989) 4-windshield, along, 11-windshield tensor(0.9988) 0-windshield, under, 1-bag tensor(0.9979) 4-windshield, along, 17-windshield tensor(0.9979) 23-windshield, under, 1-bag tensor(0.9976) 22-windshield, under, 1-bag tensor(0.9975) 8-bag, under, 19-bag tensor(0.9967) 4-windshield, along, 14-windshield tensor(0.9959) 4-windshield, along, 5-windshield tensor(0.9958) 14-windshield, along, 4-windshield tensor(0.9953) 17-windshield, under, 1-bag tensor(0.9953) 12-bag, near, 20-bag tensor(0.9951) 16-windshield, under, 1-bag tensor(0.9943) 8-bag, under, 1-bag tensor(0.9935) 4-windshield, along, 6-windshield tensor(0.9930) 26-windshield, under, 19-bag tensor(0.9927) 6-windshield, under, 15-windshield tensor(0.9921) 21-hair, under, 1-bag tensor(0.9910) 11-windshield, under, 19-bag tensor(0.9906) 6-windshield, under, 19-bag tensor(0.9904) 4-windshield, along, 2-windshield tensor(0.9900) 15-windshield, under, 1-bag tensor(0.9895) 22-windshield, under, 19-bag tensor(0.9893) 22-windshield, under, 8-bag tensor(0.9887) 26-windshield, under, 1-bag tensor(0.9879) 8-bag, under, 22-windshield tensor(0.9876) 15-windshield, under, 19-bag tensor(0.9870) 0-windshield, under, 4-windshield tensor(0.9870)

waxnkw commented 1 year ago

Sorry for the unclearness. It does not matter where you put your checkpoint. Please indicate the path to the weight at "./visualization/demo/last_checkpoint" file. For example, you put it at "/home/zhangao/ckpt.pth". You can just write this path in "./visualization/demo/last_checkpoint".