Skip to content

自然场景分类,分别使用vgg16、resnet、senet(res50)、densenet进行尝试

Notifications You must be signed in to change notification settings

whudonggu/classification

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

classification

自然场景分类,分别使用vgg16、resnet、senet(res50)、densenet进行尝试

trynet.py负责对模型进行训练及预测

model.py负责设置使用模型及部分超参数

vgg16.py搭建vgg16网络

resnet_with_senet.py搭建resnet网络并可设置是否使用SE

densenet.py搭建desnet169网络

scene_input.py和im2tf.py负责数据读入,分别对应图像格式及TFRecords格式

About

自然场景分类,分别使用vgg16、resnet、senet(res50)、densenet进行尝试

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%