Comments (4)
Thank you very much! This solves my problem.
from raster-vision.
Can you print out class_config
at the point in your code where you save the labels? Also pred_labels.pixel_scores.shape
.
from raster-vision.
This is the print of my class_config:
names=['No Data', 'Water', 'Wetlands', 'Tree Canopy', 'Shrubland', 'Low Vegetation', 'Barren', 'Structures', 'Impervious Surfaces', 'Impervious Roads', 'Tree Canopy over Structures', 'Tree Canopy over Impervious Surfaces', 'Tree Canopy over Impervious Roads', 'Aberdeen Proving Ground', 'null'] colors=[(0, 0, 0), (0, 197, 255), (0, 168, 132), (38, 115, 0), (76, 230, 0), (163, 255, 115), (255, 170, 0), (255, 0, 0), (156, 156, 156), (0, 0, 0), (115, 115, 0), (230, 230, 0), (255, 255, 115), (197, 0, 255), (71, 3, 220)] null_class='null'
When I make print(pred_labels.pixel_scores.shape)
:
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
Cell In[21], line 1
----> 1 pred_labels.pixel_scores.shape
AttributeError: 'SemanticSegmentationDiscreteLabels' object has no attribute 'pixel_scores'
from raster-vision.
Looks like an extra null class is being added to the ClassConfig
by RGBClassTransformer
, which is messing things up. Since you already have a 'No Data'
class, I would recommend passing null_class='No data'
when defining your ClassConfig
(see docs).
from raster-vision.
Related Issues (20)
- Issues with using model bundle for prediction HOT 15
- Cannot import ClassConfig on Kaggle HOT 16
- Improve unit test coverage of CLI and `Runner`s
- Cannot plot batch with ObjectDetectionVisualizer HOT 4
- Multi-temporal raster source visualizer fails when batch size is 1 HOT 2
- Make it possible to exclude "null" class labels from the computation of metrics HOT 3
- RuntimeError: expected scalar type Long but found Int HOT 10
- Allow user to specify AOI box filtering behavior in sliding window datasets HOT 1
- self._hds cannot be converted to a Python object for pickling HOT 2
- Semantic Segmentation Labels not initializing properly from predictions when extent provided HOT 2
- use my trained modle to prediction ,has wrong happened HOT 2
- RuntimeError: The size of tensor a (82) must match the size of tensor b (64) at non-singleton dimension 3 HOT 4
- Migrate to `pydantic` v2
- MPL notice for use of everett library and LGPL for triangle
- v0.30 release checklist
- `ModuleNotFoundError: No module named 'rastervision.examples'` when running command from examples doc HOT 1
- Add ability to use different Objectdetection models than FasterRCNN HOT 1
- BATCH_CPU_JOB_QUEUE requires a value parseable by str HOT 2
- Compute RasterStats from transformed RasterSource HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from raster-vision.