kirigaya/Slide-SAM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
transcendentsky 2024-03-20 22:07:35 +08:00
parent 88745c6105
commit ce31f13be0
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
datasets/cache_dataset3d.py
View File

@ -262,6 +262,8 @@ if __name__ == "__main__":
from tutils.new.manager import ConfigManager from tutils.new.manager import ConfigManager
config = ConfigManager() config = ConfigManager()
config.add_config("configs/vit_b.yaml") config.add_config("configs/vit_b.yaml")
dataset = Dataset3D(config=config['dataset'], use_cache=True)
# Caching data
dataset = Dataset3D(config=config['dataset'], use_cache=False)
dataset.caching_data() dataset.caching_data()
# dataset.convert_masks_types() # dataset.convert_masks_types()

2
datasets/cache_dataset3d3.py
View File

@ -33,7 +33,7 @@ class Dataset3D(basic_3d_dataset):
if not os.path.isdir(dirpath): if not os.path.isdir(dirpath):
continue continue
prefix = dirpath.split("/")[-1] prefix = dirpath.split("/")[-1]
if prefix[:2] in config['cache_prefix']: if prefix.split("_")[0] in config['cache_prefix']:
data_paths += glob.glob(dirpath + "/label_jpg/*.jpg") data_paths += glob.glob(dirpath + "/label_jpg/*.jpg")
print("Load ", dirpath) print("Load ", dirpath)
print('Masks len {}'.format(len(data_paths))) print('Masks len {}'.format(len(data_paths)))

2
test/volume_eval.py
View File

@ -141,7 +141,7 @@ if __name__ == "__main__":
'dataset_list': ['word'], # ["sabs"], chaos, word 'dataset_list': ['word'], # ["sabs"], chaos, word
'label_idx': 1, 'label_idx': 1,
}, },
'pth': "model_latest.pth" 'pth': "model.pth"
} }
config = ConfigManager() config = ConfigManager()