src/config.py

import torch
import torchvision

from src.models import MobileNetV3Large


class Config:
    # - - - DATA - - - #
    classes = [
        'african buffalo', 'african elephant', 'alligator', 'alligators', 'alpine ibex', 'anaconda', 'antelope', 'badger', 'bat', 'bear', 'beavers', 'bee', 'beetle', 'bighorn sheep', 'bison', 'black bear', 'blue whale', 'boar', 'bottlenose dolphin', 'bowhead whale', 'butterfly', 'capybara', 'cat', 'caterpillar', 'catfish', 'chamois', 'cheetah', 'chicken', 'chimpanzee', 'clownfish', 'cockroach', 'common dolphin', 'cow', 'coyote', 'crab', 'crayfish', 'crocodile', 'crocodiles', 'crow', 'deer', 'dog', 'dolphin', 'donkey', 'dragonfly', 'duck', 'eagle', 'elephant', 'elk', 'false killer whale', 'flamingo', 'fly', 'fox', 'frogs', 'gazelle', 'geese', 'giraffe', 'goat', 'golden eagle', 'goldfish', 'goose', 'gorilla', 'grasshopper', 'gray whale', 'hamster', 'hare', 'hedgehog', 'himalayan tahr', 'hippopotamus', 'hornbill', 'horse', 'hummingbird', 'humpback dolphin', 'humpback whale', 'hyena', 'irrawaddy dolphin', 'jaguar', 'jellyfish', 'kangaroo', 'koala', 'ladybugs', 'leopard', 'lion', 'lizard', 'lobster', 'lynx', 'macaw', 'marmot', 'minke whale', 'monkey', 'moose', 'mosquito', 'moth', 'mountain goat', 'mouse', 'octopus', 'okapi', 'orangutan', 'orca', 'ostrich', 'otter', 'owl', 'ox', 'oyster', 'panda', 'panther', 'parrot', 'pelecaniformes', 'penguin', 'pig', 'pigeon', 'pika', 'poison dart frog', 'porcupine', 'possum', 'prairie dog', 'rabbit', 'raccoon', 'rat', 'red panda', 'reindeer', 'rhino', 'rhinoceros', 'right whale', 'rocky mountain elk', 'salmon', 'sandpiper', 'seahorse', 'seal', 'sei whale', 'shark', 'sheep', 'sloth', 'snails', 'snake', 'snow leopard', 'sparrow', 'sperm whale', 'spotted hyena', 'squid', 'squirrel', 'starfish', 'swan', 'tapir', 'tiger', 'toads', 'toucan', 'turkey', 'turtle', 'warthog', 'water snakes', 'whale', 'wildebeest', 'wolf', 'wolverine', 'wombat', 'woodpecker', 'zebra'
        ]

    num_classes = len(classes)

    # - - - PARAMS - - - #
    modelpath = './src/models/mobilenetv3_large_100_224.pt'
    imgsize = 224
    model = MobileNetV3Large(modelpath, num_classes)