mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-06 11:00:48 +00:00
31 lines
660 B
Python
31 lines
660 B
Python
from collections import OrderedDict
|
|
|
|
|
|
class __ImageDisplMixin:
|
|
def displ_item(self, index):
|
|
sample, ann = self.__getitem__(index), self.annotation[index]
|
|
|
|
return OrderedDict(
|
|
{
|
|
"file": ann["image"],
|
|
"caption": ann["caption"],
|
|
"image": sample["image"],
|
|
}
|
|
)
|
|
|
|
|
|
class __AudioDisplMixin:
|
|
def displ_item(self, index):
|
|
sample, ann = self.__getitem__(index), self.annotation[index]
|
|
|
|
# TODO: Finish the Audio Display Mixin
|
|
'''
|
|
return OrderedDict(
|
|
{
|
|
}
|
|
)
|
|
'''
|
|
|
|
raise NotImplementedError
|
|
|