连接V7Lab的Darwin和Pytorch的小包装
darwin-pytorch的Python项目详细描述
达尔文派托克
这是一个用于连接V7Lab注释工具的小软件包 Darwin与{a3}一起使用。在
快速入门
安装Pythorch
第一个install your preferred version of Pytorch,包括torchvision
。在
安装darwin Pythorch
然后使用pip安装darwin-pytorch
:
pip install darwin_pytorch
示例
加载注释
darwin-pytorch
提供了一个高级包装
^{
验证和下载带注释数据集的示例:
^{pr2}$数据集
darwin-pytorch
还提供了一个Pytorch Dataset
包装从达尔文下载的数据集。示例用法:
ds = darwin_pytorch.DarwinDataset(data_path)
ds[0] # --> (PIL.Image, target-int)
ds[:2] # --> [(PIL.Image, target-int), (PIL.Image, target-int))]
ds[[0, 3]] # --> [(PIL.Image, target-int), (PIL.Image, target-int)]
请注意,目前只支持二进制分类数据集。在
许可证
darwin-pytorch
是MIT风格的许可证,如许可证文件中所示。在
- 项目
标签: