自有数据集训练 - Githubissues

thuml / Transfer-Learning-Library

Transfer Learning Library for Domain Adaptation, Task Adaptation, and Domain Generalization

http://transfer.thuml.ai

MIT License

3.39k stars 553 forks source link

自有数据集训练 #170

Closed vanity1129 closed 2 years ago

vanity1129 commented 2 years ago

请问是否可以提供一些使用自有数据集的相关设置，按照目录格式组织好数据后还需要对数据集进行注册等等的操作

thucbx99 commented 2 years ago

以Domain Adaptation任务中的OfficeHome数据集为例，其实现如下

基类为ImageList，实现了读取数据的策略，具体可以参见注释
OfficeHome类的实现根据基类的需求传入了相应的超参数
注册只需要在__init__.py的属性all中加入OfficeHome即可

以上是基于ImageList实现的一个简单流程，我们的数据集都实现在datasets目录下，您可以参考其中的实现，包括更加复杂的任务如ReID、Segmentation、Detection等