Commit
·
489f48b
1
Parent(s):
3aa4a7d
ADD new model
Browse files- app.py +1 -1
- weights/nsfw_detector_e_rok.pt +3 -0
app.py
CHANGED
@@ -21,7 +21,7 @@ from utils.torch_utils import select_device
|
|
21 |
|
22 |
# YOLOv9 모델 로드
|
23 |
device = select_device('')
|
24 |
-
model = DetectMultiBackend('./weights/
|
25 |
stride, names, pt = model.stride, model.names, model.pt
|
26 |
imgsz = check_img_size((640, 640), s=stride) # check image size
|
27 |
|
|
|
21 |
|
22 |
# YOLOv9 모델 로드
|
23 |
device = select_device('')
|
24 |
+
model = DetectMultiBackend('./weights/nsfw_detector_e_rok.pt', device=device, dnn=False, data=None, fp16=False)
|
25 |
stride, names, pt = model.stride, model.names, model.pt
|
26 |
imgsz = check_img_size((640, 640), s=stride) # check image size
|
27 |
|
weights/nsfw_detector_e_rok.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:36ff9b22a32e064019b1c2c1b79bdda1a48f4abeb9183d261de93f75d1a856f6
|
3 |
+
size 139980239
|