TexasInstruments / edgeai-modelzoo 例程问题-TI中文支持网

您好，我在使用咱们公司 Github官网的例程进行训练：https://github.com/TexasInstruments/edgeai-modelzoo/tree/main/models/vision/segmentation

我训练 cityspace 官方数据集，可以进行训练，并推理。

我将自己的数据集修改为 cityspaces 格式进行训练，可以训练，但是推理出来的结果不正确，类似于马赛克的效果。

以下是我的 cityscapes_plus.py 参数设置：

"""CityscapesLoader: Data is derived from CityScapes, and can be downloaded from here: https://www.cityscapes-dataset.com/downloads/

Many Thanks to @fvisin for the loader repo: """>github.com/…/cityscapes.py"""

colors = [[128, 64, 128], [244, 35, 232], [0, 0, 0]]

label_colours = dict(zip(range(2), colors))

ignore_index = 255

void_classes = [-1, ignore_index]

valid_classes = [0, 1]

# class_names = [‘road’, ‘pudele’]

class_map = dict(zip(valid_classes, range(len(valid_classes))))

num_classes_ = len(valid_classes)

class_weights_ = np.array([0.22567085, 1.89944273], dtype=float)

我只有两个类：road（0）和puddle（1），其余为255。

问题：

1、请问，是我的参数设置错了吗？还是其他位置的赛马需要修改呀。

2、对于我自己的数据，采用的是 cityspace 官方转换代码转的，是数据转换格式的问题吗？

Links：

感谢您对TI产品的关注！为更加有效地解决您的问题，我们建议您将问题发布在E2E英文技术论坛上（ https://e2e.ti.com/p/addpost ).

TRANSLATE with x

English

Arabic
Hebrew
Polish

Bulgarian
Hindi
Portuguese

Catalan
Hmong Daw
Romanian

Chinese Simplified
Hungarian
Russian

Chinese Traditional
Indonesian
Slovak

Czech
Italian
Slovenian

Danish
Japanese
Spanish

Dutch
Klingon
Swedish

English
Korean
Thai

Estonian
Latvian
Turkish

Finnish
Lithuanian
Ukrainian

French
Malay
Urdu

German
Maltese
Vietnamese

Greek
Norwegian
Welsh

Haitian Creole
Persian

TRANSLATE with

COPY THE URL BELOW

Back

EMBED THE SNIPPET BELOW IN YOUR SITE

Enable collaborative features and customize widget: Bing Webmaster Portal
Back

TexasInstruments / edgeai-modelzoo 例程问题

Links：

相关推荐

热门标签

热门文章

QQ咨询

回顶部