This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
OpenI
/
MegEngine
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
31
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
Browse Source
fix(imperative/dataset): fix VOC dataset
GitOrigin-RevId:
a46a4c1173
release-1.4
Megvii Engine Team
4 years ago
parent
2c2beac915
commit
aa0e11febe
1 changed files
with
24 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+24
-0
imperative/python/megengine/data/dataset/vision/voc.py
+ 24
- 0
imperative/python/megengine/data/dataset/vision/voc.py
View File
@@ -184,3 +184,27 @@ class PascalVOC(VisionDataset):
"train",
"tvmonitor",
)
class_colors = [
[0, 0, 0], # background
[0, 0, 128],
[0, 128, 0],
[0, 128, 128],
[128, 0, 0],
[128, 0, 128],
[128, 128, 0],
[128, 128, 128],
[0, 0, 64],
[0, 0, 192],
[0, 128, 64],
[0, 128, 192],
[128, 0, 64],
[128, 0, 192],
[128, 128, 64],
[128, 128, 192],
[0, 64, 0],
[0, 64, 128],
[0, 192, 0],
[0, 192, 128],
[128, 64, 0],
]
Write
Preview
Loading…
Cancel
Save