有效地提取、转换数据集并将其加载到pytorch模型中
torchetl的Python项目详细描述
Torchetl
如果您正在处理分类问题,使用本机格式的数据集(jpg、bmp等)并将pytorch放在您的存储库中,您很可能会觉得datasetfolder或imagefolder不够好。vanillatorch.utils.data.dataset也一样。这个库试图通过扩展torch.utils.data.dataset来弥合这一差距,从而有效地提取、转换和加载数据。
主要特点
extract类将把数据集划分为train、validation和test csv
transformandload类将有效地转换和使用数据集
先决条件
python 3.7.2(如果支持类型检查,则其他版本可能会工作)
手电筒
火炬视野
努比
熊猫
opencv python
sklearn
或者只需下载requirements.txt并启动“pip3 install-r requirements.txt”
安装
pip3安装torchetl
教程
请参见tutorial/tutorial.ipynb