Skip to content

rudransh2004/1D-RESNET-50-101-Models-in-Pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 

Repository files navigation

Project Title

This Repo will help you to make a your own 1 dimension Resnet-50 model in Pytorch from scratch.

Dataset Used

Crema-d Audio classification dataset was used to test out the Resnet model. Refer to Jupyter Notebook to see the usage of the implemented model.

Usage/Examples

Parameters

data_channels -> Dimentions of the input data I have used 40 since I'm using MFCC features for audio

num_classes -> In this example it is 6 but you can change according to the no of classes you have in your data.

from resnet import Resnet

#1D Resnet 50 model 

def Resnet50(layers = [3,4,6,3],data_channels=40, num_classes=6):
    return Resnet(layers, data_channels, num_classes)


#1D Resnet 101 model 
def ResNet101(layers= [3,4,23,3], data_channels=40 ,num_classes=6):
    return ResNet(block, [3, 4, 23, 3], img_channel, num_classes)

#1D Resnet 152 model 
def Resnet152(layers = [3,8,36,3],data_channels=40,num_classes=6):
    return Resnet(layers,data_channels,num_classes)

Acknowledgements

Authors

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published