/AbnormalDetection

initial comit

Primary LanguagePython

AbnormalDetection

environment installation

create env file

  • You must create .env file to set environmental variables.
wandb_api_key=[Your Key] # "xxxxxxxxxxxxxxxxxxxxxxxx"
data_dir=[Your Path] # "/home/repos/open"
  • wandb_api_key is able to get from this link.
  • data_dir is the directory name where datafile is downloaded.

conda installation

conda env create -f environment.yaml
conda activate abnormal

train model

python main.py
  • If you want to run pretrained resnet50, you simply run above.
  • If you want to change the configuration, you need to make new yaml file in the config directory.