зеркало из
https://github.com/ssciwr/AMMICO.git
synced 2025-10-30 13:36:04 +02:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Этот коммит содержится в:
родитель
097142fd36
Коммит
a6ea571478
@ -22,27 +22,9 @@ def test_read_img():
|
|||||||
test_img = ms.read_img(TEST_IMAGE_2)
|
test_img = ms.read_img(TEST_IMAGE_2)
|
||||||
assert numpy.array(test_img)[257][34] == [70, 66, 63]
|
assert numpy.array(test_img)[257][34] == [70, 66, 63]
|
||||||
|
|
||||||
|
# def test_load_feature_extractor_model_blip2():
|
||||||
|
# multimodal_device = device("cuda" if cuda.is_available() else "cpu")
|
||||||
#def test_load_feature_extractor_model_blip2():
|
# (model, vis_processors, txt_processors,) = ms.load_feature_extractor_model_blip2(multimodal_device)
|
||||||
# multimodal_device = device("cuda" if cuda.is_available() else "cpu")
|
|
||||||
# (model, vis_processors, txt_processors,) = ms.load_feature_extractor_model_blip2(multimodal_device)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
mydict = {}
|
mydict = {}
|
||||||
for img_path in images:
|
for img_path in images:
|
||||||
@ -54,4 +36,4 @@ def test_read_img():
|
|||||||
keys = list(mydict.keys())
|
keys = list(mydict.keys())
|
||||||
assert len(mydict) == 12
|
assert len(mydict) == 12
|
||||||
for key in keys:
|
for key in keys:
|
||||||
assert len(mydict[key]["3_non-deterministic summary"]) == 3
|
assert len(mydict[key]["3_non-deterministic summary"]) == 3
|
||||||
|
|||||||
Загрузка…
x
Ссылка в новой задаче
Block a user