Commit
·
1ebf67c
1
Parent(s):
0e8f091
perf: using different model
Browse files
utils/api.py
CHANGED
@@ -11,7 +11,7 @@ from .torch_utils import select_device
|
|
11 |
|
12 |
# 모델 로드
|
13 |
device = select_device('')
|
14 |
-
model = DetectMultiBackend('weights/
|
15 |
names = model.names
|
16 |
imgsz = (640, 640)
|
17 |
|
|
|
11 |
|
12 |
# 모델 로드
|
13 |
device = select_device('')
|
14 |
+
model = DetectMultiBackend('weights/yolov9_c_nsfw.pt', device=device, dnn=False, data='data/coco128.yaml', fp16=False)
|
15 |
names = model.names
|
16 |
imgsz = (640, 640)
|
17 |
|
weights/nsfw_detector_e_rok.pt
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:36ff9b22a32e064019b1c2c1b79bdda1a48f4abeb9183d261de93f75d1a856f6
|
3 |
-
size 139980239
|
|
|
|
|
|
|
|
weights/{nsfw_detector_rok.pt → yolov9_c_nsfw.pt}
RENAMED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 102830575
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0ba78c621334093b8c87b4a22559723b8ae3fbc091a5a0018e5e04602a943877
|
3 |
size 102830575
|