How can i train 2 classes model ?

chenjun2hao / DDRNet.pytorch

This is the unofficial code of Deep Dual-resolution Networks for Real-time and Accurate Semantic Segmentation of Road Scenes. which achieve state-of-the-art trade-off between accuracy and speed on cityscapes and camvid, without using inference acceleration and extra data

Other

157 stars 42 forks source link

How can i train 2 classes model ? #21

Open Yaoxingtian opened 2 years ago

Yaoxingtian commented 2 years ago

i want to train only two classes. one class + background , can you pls. tell me how i modify the code ? thanks!

chenjun2hao commented 2 years ago

这是来自QQ邮箱的假期自动回复邮件。您好，我最近正在休假中，无法亲自回复您的邮件。我将在假期结束后，尽快给您回复。

Killuaisaack114514 commented 2 months ago

Hello!Have you found the solution?I also want to train only two classes.