Skip to content

Mannix1994/MVCNN-Keras

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HI

this is the MVCNN model coding with keras.
Inspired by WeiTang114's project MVCNN-TensorFlow.

Requirements

  • CUDA 9.0 (if you have NVIDIA GPU)
  • python 2.7 or python 3.5+
  • tensorflow 1.12.0
  • tensorflow-gpu 1.12.0 (if you have NVIDIA GPU)
  • nvidia-ml-py(for python 2.7)
  • nvidia-ml-py3(for python 3.5+)
  • some other python packages

Dataset

In the original MVCNN project page

Train

# for having zero gpu or one gpu
python train.py
# for having more than one gpu
python train.py -u

Evaluate

pthon evaluate.py

Predict

python predict.py

Note

there still have some overfitting problem

About

MVCNN Implementation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages