The pytorch model and saved weights for my paper https://dl.acm.org/doi/abs/10.1145/3373017.3373039
pytorch model files
- Resnet_base_50_model.py
- Resnet_base_50_parts.py
training image of sample [3 class] [the picture is watermarked temporarily]
- ./train_image_sample_3c/*
training model file[for the first stage of our two stage model]