22 lines
499 B
Python
22 lines
499 B
Python
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
from __future__ import division
|
|
|
|
import sys
|
|
import os
|
|
import os.path as osp
|
|
import torchreid
|
|
from torchreid.data.datasets.video.FRIDA import FRIDA
|
|
from torchreid.data.datasets.image.FRIDAimg import FRIDAimg
|
|
|
|
try:
|
|
torchreid.data.register_video_dataset('FRIDA', FRIDA)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
# try:
|
|
# torchreid.data.register_image_dataset('FRIDAimg', FRIDAimg)
|
|
# except Exception as e:
|
|
# print(e) |