зеркало из
https://github.com/ssciwr/AMMICO.git
synced 2025-11-01 22:46:06 +02:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Этот коммит содержится в:
родитель
097142fd36
Коммит
a6ea571478
@ -22,28 +22,10 @@ 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():
|
# def test_load_feature_extractor_model_blip2():
|
||||||
# multimodal_device = device("cuda" if cuda.is_available() else "cpu")
|
# multimodal_device = device("cuda" if cuda.is_available() else "cpu")
|
||||||
# (model, vis_processors, txt_processors,) = ms.load_feature_extractor_model_blip2(multimodal_device)
|
# (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:
|
||||||
id_ = os.path.splitext(os.path.basename(img_path))[0]
|
id_ = os.path.splitext(os.path.basename(img_path))[0]
|
||||||
|
|||||||
Загрузка…
x
Ссылка в новой задаче
Block a user