mrwu-mac / DIFNet

[CVPR 2022] This repository is for the paper ``DIFNet: Boosting Visual Information Flow for Image Captioning'' .
BSD 3-Clause "New" or "Revised" License
20 stars 7 forks source link

代码运行问题 #14

Open Leng-bingo opened 1 year ago

Leng-bingo commented 1 year ago

我的参数如下,运行时出现特征找不到的情况,只能找到部分。 错误如下 UserWarning: Could not find detections for 280347 warnings.warn('Could not find detections for %d' % image_id) Epoch 0 - train: 1%|▊ | 68/11328 [00:52<2:25:00, 1.29it/s, loss=7.8]

    parser = argparse.ArgumentParser(description='DIFNet')
    parser.add_argument('--exp_name', type=str, default='DIFNet')

    parser.add_argument('--batch_size', type=int, default=50)
    parser.add_argument('--workers', type=int, default=4)
    parser.add_argument('--m', type=int, default=40)
    parser.add_argument('--head', type=int, default=8)
    parser.add_argument('--warmup', type=int, default=10000)
    parser.add_argument('--resume_last', action='store_true')
    parser.add_argument('--resume_best', action='store_true')
    parser.add_argument('--features_path', type=str, default='/root/qcp/Leng/DIF/coco_grid_feats/X152_grid_feats_coco_trainval.hdf5')
    parser.add_argument('--pixel_path', type=str, default='./segmentations-2')
    parser.add_argument('--annotation_folder', type=str, default='./annotations')
    parser.add_argument('--logs_folder', type=str, default='./output/tensorboard_logs')
    parser.add_argument('--model_path', type=str, default='./output/saved_transformer_models')

    parser.add_argument('--mode', type=str, default='difnet', choices=['base', 'base_lrp', 'difnet', 'difnet_lrp'])
Leng-bingo commented 1 year ago

已解决,打扰了

zdpaa commented 3 months ago

请问怎么解决上面的问题的?