关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>
magic_liu
/
Swin_pretrain
Not watched
1
0
Fork 0
Code
Releases 0 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Pull Requests
/
Detail
Labels Milestones
New Pull Request

#10 更新 'data/build.py'

Merged
magic_liu merged 1 commits from magic_liu-patch-5 into master 1 year ago
Conversation 0 Commits 1 Files Changed 1
1 changed files with 3 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +3
    -1
      data/build.py

+ 3
- 1
data/build.py View File

@@ -109,7 +109,9 @@ def build_dataset(is_train, config):
# dataset_train = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'images'), txt_path=os.path.join(config.DATA.DATA_PATH, 'images', 'images.txt'), transform=transform)
# dataset_val = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'val'), txt_path=os.path.join(config.DATA.DATA_PATH, 'val', 'val.txt'), transform=transform)
# dataset = torch.utils.data.ConcatDataset([dataset_train, dataset_val])
dataset = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'train'), txt_path=os.path.join(config.DATA.DATA_PATH, 'p2_train.txt'), transform=transform)
dataset_train = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'train'), txt_path=os.path.join(config.DATA.DATA_PATH, 'p2_train.txt'), transform=transform)
dataset_val = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'test'), txt_path=os.path.join(config.DATA.DATA_PATH, 'p2_val.txt'), transform=transform)
dataset = torch.utils.data.ConcatDataset([dataset_train, dataset_val])
else:
# dataset = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'val'), txt_path=os.path.join(config.DATA.DATA_PATH, 'val', 'val.txt'), transform=transform)
dataset = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'test'), txt_path=os.path.join(config.DATA.DATA_PATH, 'p2_val.txt'), transform=transform)


Write Preview
Loading…
Cancel
Save
Please read the following content carefully:

Dear OpenI User

Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.

For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》

Agree and continue
Disagree, exit