leensoo
/
AutoPercenty2
forked from ckh08045/AutoPercenty
1
0
Fork 0
Code Pull Requests Activity

Test Committ

This commit is contained in:
ckh08045 2024-03-26 23:10:14 +09:00
parent 6b7ffc4d59
commit d6c668d6a8
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ai/compare.py
View File

@ -58,6 +58,7 @@ feature_model.eval()
# detection_model.eval()``
# 사전 학습된 모델을 불러옵니다.
# feature_model = models.resnet18(weights=models.ResNet18_Weights.DEFAULT).to(device)
# feature_model = models.resnet50(weights=models.ResNet50_Weights.DEFAULT).to(device)