Skip to content

Commit

Permalink
Add model to list, Fix url
Browse files Browse the repository at this point in the history
  • Loading branch information
kyakuno committed Aug 9, 2024
1 parent 40b7622 commit d5f598a
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 1 deletion.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,7 @@ The collection of pre-trained, state-of-the-art AI models.
| [<img src="anomaly_detection/padim/output.png" width=128px>](anomaly_detection/padim/) | [padim](/anomaly_detection/padim/) | [PaDiM-Anomaly-Detection-Localization-master](https://github.com/xiahaifeng1995/PaDiM-Anomaly-Detection-Localization-master) | Pytorch | 1.2.6 and later | [EN](https://medium.com/axinc-ai/padim-a-machine-learning-model-for-detecting-defective-products-without-retraining-5daa6f203377) [JP](https://medium.com/axinc/padim-%E5%86%8D%E5%AD%A6%E7%BF%92%E4%B8%8D%E8%A6%81%E3%81%A7%E4%B8%8D%E8%89%AF%E5%93%81%E6%A4%9C%E7%9F%A5%E3%82%92%E8%A1%8C%E3%81%86%E6%A9%9F%E6%A2%B0%E5%AD%A6%E7%BF%92%E3%83%A2%E3%83%87%E3%83%AB-69add653fbd3) |
| [<img src="anomaly_detection/spade-pytorch/output.png" width=128px>](anomaly_detection/spade-pytorch/) | [spade-pytorch](/anomaly_detection/spade-pytorch/) | [Sub-Image Anomaly Detection with Deep Pyramid Correspondences](https://github.com/byungjae89/SPADE-pytorch) | Pytorch | 1.2.6 and later | |
| [<img src="anomaly_detection/patchcore/output.png" width=128px>](anomaly_detection/patchcore/) | [patchcore](/anomaly_detection/patchcore/) | [PatchCore_anomaly_detection](https://github.com/hcw-00/PatchCore_anomaly_detection) | Pytorch | 1.2.6 and later | |
| [<img src="anomaly_detection/mahalanobisad/bottle_test_good_000.pngg" width=128px>](anomaly_detection/mahalanobisad/) | [mahalanobisad](/anomaly_detection/mahalanobisad/) | [MahalanobisAD-pytorch](https://github.com/byungjae89/MahalanobisAD-pytorch/tree/master) | Pytorch | 1.2.9 and later | |

## Audio processing

Expand Down
2 changes: 1 addition & 1 deletion anomaly_detection/mahalanobisad/mahalanobisad.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,7 @@
# ==========================
MODEL_PATH = "efficientnet-" + args.model + ".onnx.prototxt"
WEIGHT_PATH = "efficientnet-" + args.model + ".onnx"
REMOTE_PATH = "https://storage.googleapis.com/ailia-models/efficientnet/"
REMOTE_PATH = "https://storage.googleapis.com/ailia-models/mahalanobisad/"


# ======================
Expand Down
1 change: 1 addition & 0 deletions scripts/download_all_models.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ cd ../../action_recognition/driver-action-recognition-adas; python3 driver-actio
cd ../../anomaly_detection/padim; python3 padim.py ${OPTION}
cd ../../anomaly_detection/spade-pytorch; python3 spade-pytorch.py ${OPTION}
cd ../../anomaly_detection/patchcore; python3 patchcore.py ${OPTION}
cd ../../anomaly_detection/mahalanobisad; python3 mahalanobisad.py ${OPTION}
cd ../../audio_processing/crnn_audio_classification; python3 crnn_audio_classification.py ${OPTION}
cd ../../audio_processing/deepspeech2; python3 deepspeech2.py ${OPTION}
cd ../../audio_processing/pytorch-dc-tts/; python3 pytorch-dc-tts.py ${OPTION}
Expand Down

0 comments on commit d5f598a

Please sign in to comment.