MSE-CNN Implementation 1
Code database with the implementation of MSE-CNN, from the paper 'DeepQTMT: A Deep Learning Approach for Fast QTMT-based CU Partition of Intra-mode VVC'
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
msecnn_raulkviana.custom_dataset.SamplerStg5 Class Reference
Inheritance diagram for msecnn_raulkviana.custom_dataset.SamplerStg5:

Public Member Functions

 __init__ (self, data_source, batch_size)
 Args:
 
 __iter__ (self)
 
 __len__ (self)
 

Public Attributes

 data_source
 
 batch_size
 

Constructor & Destructor Documentation

◆ __init__()

msecnn_raulkviana.custom_dataset.SamplerStg5.__init__ (   self,
  data_source,
  batch_size 
)

Args:

Parameters
data_source(Dataset): dataset to sample from
batch_size(int): Batch size to sample data

Member Function Documentation

◆ __iter__()

msecnn_raulkviana.custom_dataset.SamplerStg5.__iter__ (   self)

◆ __len__()

msecnn_raulkviana.custom_dataset.SamplerStg5.__len__ (   self)

Member Data Documentation

◆ batch_size

msecnn_raulkviana.custom_dataset.SamplerStg5.batch_size

◆ data_source

msecnn_raulkviana.custom_dataset.SamplerStg5.data_source

The documentation for this class was generated from the following file: