提交 74896c0d 编写于 作者: J jrzaurin

modified tests dir to it can run in one command

上级 71a9b559
import numpy as np
import pandas as pd
import os
import pytest
from pytorch_widedeep.preprocessing import ImagePreprocessor
full_path = os.path.realpath(__file__)
path = os.path.split(full_path)[0]
df = pd.DataFrame({'galaxies': ['galaxy1.png', 'galaxy2.png']})
img_col = 'galaxies'
imd_dir = 'images'
imd_dir = os.path.join(path,'images')
processor = ImagePreprocessor()
X_imgs = processor.fit_transform(df, img_col, img_path=imd_dir)
###############################################################################
# There is not much to test here, since I only resize.
###############################################################################
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册