訓練方式的請教 - Githubissues

MVIG-SJTU / AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System

http://mvig.org/research/alphapose.html

Other

7.94k stars 1.97k forks source link

訓練方式的請教 #157

Closed chenloveheimei closed 5 years ago

chenloveheimei commented 5 years ago

我想請問下，你之前非實時那個用的是tensorflow，這個可以實時的版本用的是pytorch，能夠把pytorch這個版本的模型轉換成tensorflow的嗎？效果一樣的嗎？剛接觸深度學習，虛心請教謝謝

Jeff-sjtu commented 5 years ago

你好，我們非實時的版本不全是tensorflow實現。tensorflow框架實現human detector，lua的torch框架實現sppe。現在的新版本是使用pytorch把整個系統整合到一起，目前還沒有轉換到tensorflow的打算，請見諒。

chenloveheimei commented 5 years ago

我想移植到android上，但是Android看了好像没有pytorch的，Android能使用TensorFlow.所以我想就问下如果pytorch转TensorFlow是否可行而已？是否会效果一样，因为我还没接触过转换训练方法这一块，看看以你们接触这么久的经验，问下可不可行。谢谢

Jeff-sjtu commented 5 years ago

可行，但需要自行重写模型，然后把tensorflow整合到现有的AlphaPose系统中，工作量会比较大。模型部分可以看一下现有的一些开源工具是否满足需求，可以不重新训练模型。

chenloveheimei commented 5 years ago

好的謝謝。

chenloveheimei commented 5 years ago

請教下，duc_se.pth和yolov3-spp.weights分別是用來做什麼的？我在網上找到了別人用pytorch轉keras再轉tensorflow的算法，用了默認的pth是可以轉的，但是我直接用你那個duc_se.pth就會報錯，類似Missing key(s) in state_dict: "features.0.bias", "features.0.weight", "features.3.squeeze.bias"。有什麼可以提供的idea供我查下資料嗎？網站：https://cloud.tencent.com/developer/article/1166860 謝謝