Method size

lesson6-Segmentation/SegDataset.h:19–21

Return the length of data

    torch::data::Example<> get(size_t index) override;
    // Return the length of data
    torch::optional<size_t> size() const override {
        return list_labels.size();
    };
private:
    void draw_mask(std::string json_path, cv::Mat &mask);
    int resize_width = 512; int resize_height = 512;