water_turbidity_det/main.py

import torch
from torchvision.models import resnet50, ResNet50_Weights