LearningnRunning commited on
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/nsfw_detector_e_rok.pt', device=device, dnn=False, data='data/coco128.yaml', fp16=False)
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:96e4b0dd3cc2e57a5a5b8cba15e887d5196883d650feab209950a0fdbb0daf0f
3
  size 102830575
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ba78c621334093b8c87b4a22559723b8ae3fbc091a5a0018e5e04602a943877
3
  size 102830575