AlphaPose icon indicating copy to clipboard operation
AlphaPose copied to clipboard

Published 20 hours ago • verified publisher icon MVIG-SJTU

Change backbone to mobilenetv3

Open dekathomas opened this issue 2 years ago • 2 comments

Hello @Fang-Haoshu @HaoyiZhu, I have tried to update the torchvision and using mobilenet to train alphapose, bu I got error like this image

I change file alphpapose/models/fastpose.py at line 29 to self.preact = eval(f"tm.mobilenetv3_large()") and 34 x = eval(f"tm.mobilenetv3_large()")

Could you help me? Thank you in advance

dekathomas avatar Mar 23 '22 14:03 dekathomas

Could you help me with this problem please? @Fang-Haoshu @HaoyiZhu @Jeff-sjtu

dekathomas avatar Mar 29 '22 15:03 dekathomas

I'm not familiar with mobilenet, but it seems the dim of input to pixel_shuffle is wrong. Could you print it's shape out?

HaoyiZhu avatar Jun 21 '22 08:06 HaoyiZhu