Skip to content

Using DenseNet121 for very high resolution scene classification

Notifications You must be signed in to change notification settings

BiQiWHU/DenseNet121-VHRRSI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Readme In this project, we use DenseNet121,169,264 to implement very high resolution remote sensing image scene classification.

The dataset is UC Merced dataset so that in the fully connected layer and softmax classifier, the number of parameter is set to be 21.

Note that since the network structure is extremely large, here we can use the L2 regularization to fight against the over-fitting problem. (refer to line 27-29 and line 124 in DenseRS.py)

The process is listed as follows. Step1. copy UC Merced dataset into the file folder and extract it.

Step2. run tfdata.py to generate training data and test data (The format is tfrecord).

Step3. run DenseRS.py to train the model. Note that you can adjust the hyper-parameters according to your own dataset.

Step4. run test.py to test the model on the test data.

For any other question, please connact [email protected]

Enjoy!

About

Using DenseNet121 for very high resolution scene classification

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages