Spaces:
Sleeping
Sleeping
Sreekanth Tangirala
commited on
Commit
·
0601512
1
Parent(s):
98c0ee5
removed model
Browse files
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 |
-

|
45 |
-

|
46 |
-

|
47 |
-
|
48 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|