NeuralBabyTalk
NeuralBabyTalk copied to clipboard
Pytorch code of for our CVPR 2018 paper "Neural Baby Talk"
python main.py --path_opt cfgs/normal_coco_res101.yml --batch_size 20 --cuda True --num_workers 20 --max_epoch 30 but failed , file is not a zip file how to solve?
I just want to run this on new images to see the results. I am using the evaluation section and getting this error: `ImportError: /home/nk/NeuralBabyTalk/pooling/roi_align/_ext/roi_align/_roi_align.so: undefined symbol: _Py_Dealloc`
Traceback (most recent call last): File "main.py", line 213, in dataset = DataLoader(opt, split='train') File "/workspace/neuralbabytalk/misc/dataloader_coco.py", line 112, in __init__ self.dataloader_hdf = HDFSingleDataset(self.opt.proposal_h5) File "/workspace/neuralbabytalk/misc/dataloader_hdf.py", line 59, in __init__ super().__init__(...
File "main.py", line 18, in from misc import utils, eval_utils, AttModel File "/home/qian/ysk/NeuralBabyTalk/misc/AttModel.py", line 10, in from misc.model import AttModel File "/home/qian/ysk/NeuralBabyTalk/misc/model.py", line 14, in from pooling.roi_align.modules.roi_align import RoIAlignAvg File...
I prepared the data and trained Coco exactly according to your MD file. Then demo.py showed the test dataset that only 10 of the 5,000 had boxes. I found det_ix...
Hi Jiansen, Thanks for the comprehensive repository you set up! I'm in the process of trying to use NBT on a _custom_ annotated dataset. To my understanding, this is not...
thank you for releasing the repo!. Would it be possible to use this code on a video clip? could you please guide me to the changes required to ```demo.py``` to...
We tried to fix this problem, there's no key named fg_mask in the dict, someone said on the internet that this might be due to the version difference of torch,...
in the README, click the part of link, but they are not open.
According to eq(8), **c_t** is the current cell state, but in your code, you use the last cell state, so which one is right? Can you solve my question? Thanks.