Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

process_dataset in extract_cnn_features -- is it a bug? #5

Open
micklexqg opened this issue Nov 23, 2016 · 0 comments
Open

process_dataset in extract_cnn_features -- is it a bug? #5

micklexqg opened this issue Nov 23, 2016 · 0 comments

Comments

@micklexqg
Copy link

micklexqg commented Nov 23, 2016

Hi,
I believe there is a bug in the following function:
`
def process_dataset(dataset, net, gpu_id):
data_dir = paths.dataset_dir[dataset] + '/'
images_dir = paths.images_dir[dataset]
data = json.load(open(data_dir + 'dataset_%s.json' % dataset, 'r'))

splits = defaultdict(list)
for im in data['images']:
    split = im['split']
    if split == 'restval':
        split = 'train'
    splits[split].append(images_dir[dataset] + im['filepath'] + '/' + im['filename'])

`

I evaluate process_dataset('coco', 'VGG19', 1), and "string indices must be integers, not str" error appear at the line "splits[split].append(images_dir[dataset] + im['filepath'] + '/' + im['filename'])".
I debug and find maybe it has something wrong with images_dir[dataset].
@ivendrov, is it a bug?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant