search

xiaochus / MobileNetV3

A Keras implementation of MobileNetV3.
MIT License
236 stars 83 forks source link

Implementing Semantic Segmentation Head? #2

Open fzyzcjy opened 5 years ago

fzyzcjy commented 5 years ago

Firstly, thanks for the implementation of the MobileNetV3!

Do you have an interest in implementing the semantic segmentation head described in section 6.4? I am currently planning to implement that, so if you have an interest, we can work together to do it!

xiaochus commented 5 years ago

@fzyzcjy Great idea, i'd like to work with you, but i don't have time in workday. You can fork my repository and write your code firstly, i will do it in my free time.

fzyzcjy commented 5 years ago

@xiaochus OK I will try that!

fzyzcjy commented 5 years ago

How do you suggest me on testing? My naive idea: 1. Train the classification (i.e. your existing code) on ImageNet 2. Implement segmentation, and train on Cityscapes, with weights initialized from ImageNet Is that feasible, and what is your idea? In addition, do you have pretrained weights, or should I train by myself and upload that?

xiaochus commented 5 years ago

@fzyzcjy I don't have a pre-train model in imagenet. You can test the performance of mobilenetV3 in imagenet and it will be helpful in segmentation task.

fzyzcjy commented 5 years ago

OK I will make some attempt tonight

xiaochus commented 5 years ago

@fzyzcjy I add the code of Lite R-ASPP Semantic Segmentation head and you can check if i have a correct understanding of the description in the paper.

fzyzcjy commented 5 years ago

@xiaochus Oh I also did that since yesterday! But my code is currently still a big mess... Maybe I just post a gist

fzyzcjy commented 5 years ago

https://gist.github.com/fzyzcjy/25a60fa3b53f63a1b9c87f84d3e25908

noreenanwar commented 4 years ago

https://gist.github.com/fzyzcjy/25a60fa3b53f63a1b9c87f84d3e25908

have u write code for 6.4?