Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2018-11-21 18:42:06 +01:00
parent f1a94abafa
commit b0b19b3b94
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
detect.py
View File

@ -25,7 +25,7 @@ parser.add_argument('-img_size', type=int, default=32 * 13, help='size of each i
opt = parser.parse_args() opt = parser.parse_args()
print(opt) print(opt)
def detect(opt): def main(opt):
os.system('rm -rf ' + opt.output_folder) os.system('rm -rf ' + opt.output_folder)
os.makedirs(opt.output_folder, exist_ok=True) os.makedirs(opt.output_folder, exist_ok=True)
@ -142,4 +142,4 @@ def detect(opt):
if __name__ == '__main__': if __name__ == '__main__':
torch.cuda.empty_cache() torch.cuda.empty_cache()
detect(opt) main(opt)