Simplified inference (#1153)
This commit is contained in:
1 parent
c67e72200e
commit
3b57cb5641
7 files changed
+87
-30
No files matched your search
+3
-6
@@ -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
|
||||
|
||||
|
||||
|
||||
Reference in new issue
Block a user