RCRS_ML_Model/model_list.txt at master · devswha/RCRS_ML_Model · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
###########################################################
# CNN type of ML model
###########################################################
ResNet_50
ResNet_101
ResNet_152
ResNeXt_50
ResNeXt_101
Inception_v3
InceptionResNet_v2
MobileNet
DenseNet_121
DenseNet_169
DenseNet_201
DenseNet_264
Xception

###########################################################
# Attention Module
###########################################################
SE
CBAM
GCBAM

###########################################################
# Video-based
###########################################################
LSTM

** You should input the name of ML model(MODEL_TRAIN/TEST_NAME) in the config_image.txt and config_video.txt.
** The name of model should figure to [CNN type of ML model]_[Attention Module]_[Video-based].
** [Video-based] and [Attention Module] is a option.
** Example: Inception_v3, InceptionResNet_v2_LSTM, InceptionResNet_v2_CBAM, InceptionResNet_v2_CBAM_LSTM