search

cypw / PyTorch-MFNet

MIT License
252 stars 56 forks source link

2D Model #23

Closed Esaada closed 5 years ago

Esaada commented 5 years ago

Hi, where can I find the 2D model? I don't mean the weights, I mean the code. I see only mfnet3d.py file...

Thanks.

XuhongLuo commented 5 years ago

yeah , i want to get the 2D model code too ,Thanks !

cypw commented 5 years ago

2D MF-Net is implemented by MXNet, where the code is inside the released .tar files: GoogleDrive

Or, you can simply delete the 1st dimension of all 3D convolution layers in mfnet3d.py to get the PyTorch implementation of 2D MF-Net.