Simplified inference (#1153)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-10-15 20:10:08 +02:00
1 parent c67e72200e
commit 3b57cb5641
7 files changed
+87 -30

No files matched your search

+3 -6
View File
@@ -1,6 +1,5 @@
import argparse
import glob
import json
import os
import shutil
from pathlib import Path
@@ -8,19 +7,17 @@ from pathlib import Path
import numpy as np
import torch
import yaml
from sotabencheval.object_detection import COCOEvaluator
from sotabencheval.utils import is_server
from tqdm import tqdm
from models.experimental import attempt_load
from utils.datasets import create_dataloader
from utils.general import (
coco80_to_coco91_class, check_dataset, check_file, check_img_size, compute_loss, non_max_suppression, scale_coords,
xyxy2xywh, clip_coords, plot_images, xywh2xyxy, box_iou, output_to_target, ap_per_class, set_logging)
xyxy2xywh, clip_coords, set_logging)
from utils.torch_utils import select_device, time_synchronized
from sotabencheval.object_detection import COCOEvaluator
from sotabencheval.utils import is_server
DATA_ROOT = './.data/vision/coco' if is_server() else '../coco' # sotabench data dir