| 6 | |
| 7 | |
| 8 | def get_all_sequences(dataset_cfg: DictConfig, sort_by_seq_name: bool = True): |
| 9 | if isinstance(dataset_cfg.ls_all_seqs, str): |
| 10 | # if ls_all_seqs is a string, it is the root path of sequences |
| 11 | seq_list = [d for d in os.listdir(dataset_cfg.ls_all_seqs) if osp.isdir(osp.join(dataset_cfg.ls_all_seqs, d))] |
| 12 | elif isinstance(dataset_cfg.ls_all_seqs, ListConfig): |
| 13 | # if ls_all_seqs is a ListConfig, it is the ListConfig of sequence names |
| 14 | seq_list = dataset_cfg.ls_all_seqs |
| 15 | else: |
| 16 | raise ValueError(f"Unknown ls_all_seqs type: {type(dataset_cfg.ls_all_seqs)}, ls_all_seqs is {dataset_cfg.ls_all_seqs}, which should be a string or a ListConfig") |
| 17 | return sorted(seq_list) if sort_by_seq_name else seq_list |
| 18 | |
| 19 | def list_imgs_a_sequence(dataset_cfg: DictConfig, seq: Optional[str] = None): |
| 20 | subdir = dataset_cfg.img.path.format(seq=seq) # string include {seq} |