NVIDIA/DALI

Can dali webdataset reader work out classes from folder structure?

Open

#4,146 建立於 2022年8月7日

在 GitHub 查看
 (3 留言) (0 反應) (0 負責人)C++ (670 fork)auto 404
help wanted

倉庫指標

Star
 (5,722 star)
PR 合併指標
 (PR 指標待抓取)

描述

The webdataset reader seems to require a file to define the class per image in a tar? As shown in this example: https://docs.nvidia.com/deeplearning/dali/user-guide/docs/examples/general/data_loading/dataloading_webdataset.html

The standard nvidia.dali.fn.readers.file can work out the class from the folder structure as per: https://docs.nvidia.com/deeplearning/dali/user-guide/docs/operations/nvidia.dali.fn.readers.file.html

Is it possible to get derive the label for a file from the folder that it is in with webdataset or do I need to add an extra file per sample for the class?

貢獻者指南