ModelNet

ModelNet Dataset

Flux3D.Dataset.ModelNetType
ModelNet

Base ModelNet for MN10/40.

Fields:

  • root::String - Root directory of dataset
  • path::String - Directory of dataset
  • train::Bool - Specifies the trainset
  • length::Int - Length of dataset
  • datapaths::Array - Array containing the shape and path for each datapoint
  • transform - Transform to be applied to data point
  • categories::Vector{String} - Categories to be used in dataset
  • classes_to_idx::Dict{String, UInt8} - Dict mapping from shape name to class_idx
  • idx_to_classes::Dict{UInt8, String} - Dict mapping from class_idx to shape name
source

ModelNet Constructor

ModelNet10 Dataset

Flux3D.Dataset.ModelNet10Method
ModelNet10(;kwargs...)

Returns ModelNet10 dataset.

Optional Key Arguments:

* `root::String=default_root`   - Root directory of dataset
* `train::Bool=true`            - Specifies the trainset
* `transform=nothing`           - Transform to be applied to data point.
* `categories::Vector{String}`  - Specifies the categories to be used in dataset.

Examples:

julia> dset = ModelNet10(train=false)
julia> typeof(dset[1].data) == TriMesh
source

ModelNet40 Dataset

Flux3D.Dataset.ModelNet40Method
ModelNet40(;kwargs...)

Returns ModelNet40 dataset.

Optional Key Arguments:

* `root::String=default_root`   - Root directory of dataset
* `train::Bool=true`            - Specifies the trainset
* `transform=nothing`           - Transform to be applied to data point.
* `categories::Vector{String}`  - Specifies the categories to be used in dataset.

Examples:

julia> dset = ModelNet40(train=false)
julia> typeof(dset[1].data) == TriMesh
source