https://github.com/fitushar/3dcnns_tf2modelhub
Almost all the deeplearning libraries provide ready to use 2D models with/without imagenet weights, But In the case of 3D, CNN models are not as available. This repo will contain commonly used 2D CNNs 3D implementations.
https://github.com/fitushar/3dcnns_tf2modelhub
3d-models classification cnn-classification cnn-keras segmentation tf2
Last synced: about 1 year ago
JSON representation
Almost all the deeplearning libraries provide ready to use 2D models with/without imagenet weights, But In the case of 3D, CNN models are not as available. This repo will contain commonly used 2D CNNs 3D implementations.
- Host: GitHub
- URL: https://github.com/fitushar/3dcnns_tf2modelhub
- Owner: fitushar
- License: mit
- Created: 2020-12-23T03:06:45.000Z (over 5 years ago)
- Default Branch: main
- Last Pushed: 2020-12-23T03:46:09.000Z (over 5 years ago)
- Last Synced: 2025-04-22T13:08:29.778Z (about 1 year ago)
- Topics: 3d-models, classification, cnn-classification, cnn-keras, segmentation, tf2
- Language: Python
- Homepage:
- Size: 23.4 KB
- Stars: 10
- Watchers: 1
- Forks: 2
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# 3DCNNs_TF2Modelhub
Almost all the deeplearning libraries provide ready to use 2D models with/without imagenet weights, But In the case of 3D, CNN models are not as available. This repo will contain commonly used 2D CNNs 3D implementations.
## Libraries Needed:
```ruby
* Python 3.x
* Tensorflow 2.X
* Numpy
* random
```
## 3D-Models Avaible:
### Classification:
```ruby
* Resnet_3D.py
* DenseNet_3D.py
* VGG_3D.py
* Inception_3D.py
```
### Segmentation:
```ruby
* DenseVnet3D.py -> (https://github.com/fitushar/DenseVNet3D_Chest_Abdomen_Pelvis_Segmentation_tf2/)
* Unet3D.py-> (https://github.com/fitushar/3DUnet_tensorflow2.0/)
```
### How to run
Configure the models based on your need and GPUs running capability using *config.py*:
```ruby
##-----Network Configuration----#####
NUMBER_OF_CLASSES=5
INPUT_PATCH_SIZE=(96,128,128, 1)
##------Resnet3D----####
TRAIN_NUM_RES_UNIT=3
TRAIN_NUM_FILTERS=(16, 32, 64, 128)
TRAIN_STRIDES=((1, 1, 1), (2, 2, 2), (2, 2, 2), (2, 2, 2))
TRAIN_CLASSIFY_ACTICATION=tf.nn.relu6
TRAIN_KERNAL_INITIALIZER=tf.keras.initializers.VarianceScaling(distribution='uniform')
#-------DenseNet13D----#####
# DenseNet
DENSE_NET_BLOCKS = 3
DENSE_NET_BLOCK_LAYERS = 5
DENSE_NET_INITIAL_CONV_DIM = 16
DENSE_NET_GROWTH_RATE = DENSE_NET_INITIAL_CONV_DIM // 2
DENSE_NET_ENABLE_BOTTLENETCK = False # called DenseNet-BC if ENABLE_BOTTLENETCK and COMPRESSION < 1 in paper
DENSE_NET_TRANSITION_COMPRESSION = 1.0
DENSE_NET_ENABLE_DROPOUT = True
DENSE_NET_DROPOUT = 0.5
#-------Inception3D----#####
INCEPTION_BLOCKS = 6
INCEPTION_REDUCTION_STEPS = 2
INCEPTION_KEEP_FILTERS = 128
INCEPTION_ENABLE_DEPTHWISE_SEPARABLE_CONV_SHRINKAGE = 0.333
INCEPTION_ENABLE_SPATIAL_SEPARABLE_CONV = True
INCEPTION_DROPOUT = 0.5
#---------VGG3D----####
TRAIN_CLASSIFY_USE_BN = False
```