updates
This commit is contained in:
parent
cc5e9a5a85
commit
9cc5ddd776
|
@ -1,3 +1,4 @@
|
|||
import shutil
|
||||
import argparse
|
||||
import time
|
||||
from sys import platform
|
||||
|
@ -20,8 +21,8 @@ def detect(
|
|||
webcam=False
|
||||
):
|
||||
device = torch_utils.select_device()
|
||||
os.system('rm -rf ' + output)
|
||||
os.makedirs(output, exist_ok=True)
|
||||
shutil.rmtree(output) # delete output folder
|
||||
os.makedirs(output) # make new output folder
|
||||
|
||||
# Initialize model
|
||||
model = Darknet(cfg, img_size)
|
||||
|
|
Loading…
Reference in New Issue