mrgloom 242 Report post Posted August 28, 2017 Ну так а если использовать python layer, то почему caffe? Softmax по идее можно взвесить. https://github.com/alexgkendall/SegNet-Tutorial/blob/master/Example_Models/train_segnet_sun.prototxt#L1608 Но может у них опять же кастомный layer для этого. Share this post Link to post Share on other sites
prim.ko 3 Report post Posted August 29, 2017 15 часов назад, mrgloom сказал: Ну так а если использовать python layer, то почему caffe? Softmax по идее можно взвесить. https://github.com/alexgkendall/SegNet-Tutorial/blob/master/Example_Models/train_segnet_sun.prototxt#L1608 Но может у них опять же кастомный layer для этого. Ага, помню когда я взвешивал выборку camvid, в которой 325 картинок. , ну она еще с segnet идет в комплексе, получились немного другие цифры ежели у автора. Там даже issue открыли в git, в котором у народа такие же цифры вышли. Автор молчит:) Некоторые цифры в классах на порядок отличались. Caffè - потому что она единственная собирается без боли под TX*. К тому же есть TensoRT, которые вообще песня:) Share this post Link to post Share on other sites
mrgloom 242 Report post Posted August 29, 2017 А что такое TX* ? Tegra? Ну так python layer придётся переписать на C++/Cuda он же вроде только для train годится. Share this post Link to post Share on other sites
prim.ko 3 Report post Posted August 29, 2017 13 минуты назад, mrgloom сказал: А что такое TX* ? Tegra? Ну так python layer придётся переписать на C++/Cuda он же вроде только для train годится. Да. Jetson TX. Если не вру, то python layer вообще не участвует в deploy Share this post Link to post Share on other sites
mrgloom 242 Report post Posted August 29, 2017 А точно, он же только для трейна, а аутпут в деплое это выход с сигмоид. Тогда да. Напишите про опыт с TensoRT. Share this post Link to post Share on other sites
prim.ko 3 Report post Posted August 30, 2017 12 часа назад, mrgloom сказал: Напишите про опыт с TensoRT. Есть замечательный туториал по работе с TensorRT и внутренней камерой jetsona. Там работает все без унылой OpenCV - на gstremer`e вытягивается сразу на GPU. Перепилил код под свои нужды и все замечательно. DetectNet к примеру на 2 класса выдает - 15 fps (480*360) https://github.com/dusty-nv/jetson-inference#image-segmentation-with-segnet Share this post Link to post Share on other sites
idrua 8 Report post Posted June 14, 2018 Недавно где-то читал, что на UNet очень плохо заходит multi-class (плюс дисбаланс классов). Лучше использовать UNet только для бинарной. Share this post Link to post Share on other sites
