guanjihuan/GitHub_collection_Python
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Update documentation of cnn_classification.py (#7486)

Browse Source 
* Updated documentation of cnn_classification.py for much better understanding

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update computer_vision/cnn_classification.py

Co-authored-by: Caeden Perelli-Harris <caedenperelliharris@gmail.com>

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Caeden Perelli-Harris <caedenperelliharris@gmail.com>
This commit is contained in:
harshyadavcs 2022-10-26 03:13:45 +05:30 committed by GitHub
parent 7e3dff17c5
commit 2c959a7491
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
computer_vision/cnn_classification.py
View File

@ -30,9 +30,12 @@ from tensorflow.keras import layers, models
if __name__ == "__main__": if __name__ == "__main__":
# Initialising the CNN # Initialising the CNN
# (Sequential- Building the model layer by layer)
classifier = models.Sequential() classifier = models.Sequential()
# Step 1 - Convolution # Step 1 - Convolution
# Here 64,64 is the length & breadth of dataset images and 3 is for the RGB channel
# (3,3) is the kernel size (filter matrix)
classifier.add( classifier.add(
layers.Conv2D(32, (3, 3), input_shape=(64, 64, 3), activation="relu") layers.Conv2D(32, (3, 3), input_shape=(64, 64, 3), activation="relu")
) )