610265158 / face_landmark

A simple method for face alignment based on wingloss and mutitask learning :)
Apache License 2.0
251 stars 80 forks source link

是否有可能增加landmark 點 #26

Open osamu765 opened 4 years ago

osamu765 commented 4 years ago

現在想增加兩點(左右眼瞳孔位置),也已經有 lable過的資料了 是否有辦法 修改traning code 來使landmark點 到70點??

610265158 commented 4 years ago

可以, 需要修改一些东西,但是稍微复杂一些,就是把dataloader 里面的点都对应改好,然后修改输出shape

osamu765 commented 4 years ago

所以主要會改到的地方就是 處理dataset 部分的code 跟 make_json 還有修改輸出shape? 這幾天來嘗試看看,謝謝 抱歉 tensorflow 不是很熟悉。

610265158 commented 4 years ago

我改了一个pytorch 版的

flyingmrwang commented 4 years ago

@610265158 "switch to pytorch, HAHA... don't worry it wont happen." 🤔