Add satellite time-series panoptic segmentation example with ConvTemporalAttentionNet #1210
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
|
|
Thanks for your contribution! |
HydrogenSulfate
requested changes
Sep 1, 2025
Comment on lines
26
to
71
| """ | ||
| Paddle Dataset class to load samples from the PASTIS dataset, for semantic and | ||
| panoptic segmentation. | ||
| The Dataset yields ((data, dates), target) tuples, where: | ||
| - data contains the image time series | ||
| - dates contains the date sequence of the observations expressed in number | ||
| of days since a reference date | ||
| - target is the semantic or instance target | ||
| Args: | ||
| folder (str): Path to the dataset | ||
| norm (bool): If true, images are standardised using pre-computed | ||
| channel-wise means and standard deviations. | ||
| reference_date (str, Format : 'YYYY-MM-DD'): Defines the reference date | ||
| based on which all observation dates are expressed. Along with the image | ||
| time series and the target tensor, this dataloader yields the sequence | ||
| of observation dates (in terms of number of days since the reference | ||
| date). This sequence of dates is used for instance for the positional | ||
| encoding in attention based approaches. | ||
| target (str): 'semantic' or 'instance'. Defines which type of target is | ||
| returned by the dataloader. | ||
| * If 'semantic' the target tensor is a tensor containing the class of | ||
| each pixel. | ||
| * If 'instance' the target tensor is the concatenation of several | ||
| signals, necessary to train the Parcel-as-Points module: | ||
| - the centerness heatmap, | ||
| - the instance ids, | ||
| - the voronoi partitioning of the patch with regards to the parcels' | ||
| centers, | ||
| - the (height, width) size of each parcel | ||
| - the semantic label of each parcel | ||
| - the semantic label of each pixel | ||
| cache (bool): If True, the loaded samples stay in RAM, default False. | ||
| mem16 (bool): Additional argument for cache. If True, the image time | ||
| series tensors are stored in half precision in RAM for efficiency. | ||
| They are cast back to float32 when returned by __getitem__. | ||
| folds (list, optional): List of ints specifying which of the 5 official | ||
| folds to load. By default (when None is specified) all folds are loaded. | ||
| class_mapping (dict, optional): Dictionary to define a mapping between the | ||
| default 18 class nomenclature and another class grouping, optional. | ||
| mono_date (int or str, optional): If provided only one date of the | ||
| available time series is loaded. If argument is an int it defines the | ||
| position of the date that is loaded. If it is a string, it should be | ||
| in format 'YYYY-MM-DD' and the closest available date will be selected. | ||
| sats (list): defines the satellites to use (only Sentinel-2 is available | ||
| in v1.0) | ||
| """ |
There was a problem hiding this comment.
类的docstring请移动到类名下方
| ## 参考 | ||
| 原始 PyTorch 实现：[GitHub Link](https://github.com/xxx) | ||
|
|
||
| --- |
There was a problem hiding this comment.
删除该行
Suggested change
| --- |
There was a problem hiding this comment.
请在mkdocs.yml、README.md和index.md里新增案例链接
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR 类型
新增案例：PyTorch → Paddle 模型迁移
描述
包含文件
examples/satellite_panoptic/pretrained/model.pdparamsexamples/satellite_panoptic/src/model_paddle.pyexamples/satellite_panoptic/src/dataset_paddle.pyexamples/satellite_panoptic/src/utils_paddle.pyexamples/satellite_panoptic/train.pyexamples/satellite_panoptic/validate.pyexamples/satellite_panoptic/conf/satellite_panoptic.yamlexamples/satellite_panoptic/README.md本地测试结果
关联 Issue
Close #（如有）
备注
已在本地运行通过，CI 检查通过。