Sreekanth Tangirala commited on
Commit
0601512
·
1 Parent(s): 98c0ee5

removed model

Browse files
Files changed (1) hide show
  1. Readme.md +0 -48
Readme.md DELETED
@@ -1,48 +0,0 @@
1
- # ResNet50 Model Implementation
2
-
3
- This implementation provides a customizable ResNet50 model for image classification tasks.
4
-
5
- ## Model Architecture
6
-
7
- The model uses the ResNet50 architecture, which is a deep convolutional neural network with 50 layers. Key features include:
8
-
9
- - Based on the standard ResNet50 architecture
10
- - Customizable number of output classes
11
- - Modified final fully connected layer to match the desired number of classes
12
- - Initialized from scratch (no pre-training)
13
-
14
- ## Functions
15
-
16
- ### `get_model(num_classes)`
17
- Initializes a new ResNet50 model with a custom number of output classes.
18
- - **Input**: Number of classes (integer)
19
- - **Output**: Initialized ResNet50 model
20
- - **Note**: The model is initialized without pre-trained weights
21
-
22
- ### `save_model(model, path)`
23
- Saves the model's state dictionary to a specified path.
24
- - **Input**:
25
- - model: Trained PyTorch model
26
- - path: File path for saving the model
27
-
28
- ### `load_model(num_classes, path)`
29
- Loads a previously saved model.
30
- - **Input**:
31
- - num_classes: Number of output classes
32
- - path: Path to the saved model file
33
- - **Output**: Loaded ResNet50 model
34
-
35
- ## Usage Example
36
-
37
- ```python
38
- model = get_model(num_classes=1000)
39
- save_model(model, 'model.pth')
40
- loaded_model = load_model(num_classes=1000, path='model.pth')
41
- ```
42
-
43
- ## Attachments
44
- ![Trained on AWS](https://github.com/srtangirala/resnet-training/blob/main/logs/images/aws.png)
45
- ![Logs 1](https://github.com/srtangirala/resnet-training/blob/main/logs/images/logs-1.png)
46
- ![Logs 2](https://github.com/srtangirala/resnet-training/blob/main/logs/images/logs-2.png)
47
-
48
-