Commit 61312b84 authored by Jonathan Crall's avatar Jonathan Crall Committed by Jon Crall
Browse files

wip

parent 334f4837
......@@ -4108,8 +4108,9 @@ class MixinCocoAddRemove(object):
# Defined as a global for pickle
def _lut_frame_index(index, gid):
return index.imgs[gid]['frame_index']
# TODO: add a pickled test
def _lut_frame_index(imgs, gid):
return imgs[gid]['frame_index']
class CocoIndex(object):
......@@ -4167,7 +4168,7 @@ class CocoIndex(object):
Helper for ensuring that vidid_to_gids returns image ids ordered by
frame index.
"""
return sortedcontainers.SortedSet(gids, key=partial(_lut_frame_index, index))
return sortedcontainers.SortedSet(gids, key=partial(_lut_frame_index, index.imgs))
def __init__(index):
index.anns = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment