File tree Expand file tree Collapse file tree 9 files changed +11
-9
lines changed Expand file tree Collapse file tree 9 files changed +11
-9
lines changed Original file line number Diff line number Diff line change @@ -26,7 +26,7 @@ My platform is like this:
2626## get start
2727With a pretrained weight, you can run inference on an single image like this:
2828```
29- $ python tools/demo.py --model bisenetv2 --weight-path /path/to/your/weights.pth --img-path ./example.jpg
29+ $ python tools/demo.py --model bisenetv2 --weight-path /path/to/your/weights.pth --img-path ./example.png
3030```
3131This would run inference on the image and save the result image to ` ./res.jpg ` .
3232
Original file line number Diff line number Diff line change 1- /data1 /zzy/datasets/cityscapes/gtFine/
1+ /data2 /zzy/. datasets/cityscapes/gtFine/
Original file line number Diff line number Diff line change 1- /data1 /zzy/datasets/cityscapes/leftImg8bit/
1+ /data2 /zzy/. datasets/cityscapes/leftImg8bit/
Original file line number Diff line number Diff line change 1111import cv2
1212import numpy as np
1313
14- import transform_cv2 as T
15- from sampler import RepeatedDistSampler
14+ import lib . transform_cv2 as T
15+ from lib . sampler import RepeatedDistSampler
1616
1717
1818
Original file line number Diff line number Diff line change 11
2+ import sys
3+ sys .path .insert (0 , '.' )
24import argparse
35import torch
46import torch .nn as nn
Original file line number Diff line number Diff line change 22# -*- encoding: utf-8 -*-
33
44import sys
5- sys .path .insert (0 , './lib ' )
5+ sys .path .insert (0 , '.' )
66import os
77import os .path as osp
88import logging
Original file line number Diff line number Diff line change 22# -*- encoding: utf-8 -*-
33
44import sys
5- sys .path .insert (0 , './lib ' )
5+ sys .path .insert (0 , '.' )
66import os
77import os .path as osp
88import random
2020from lib .models import model_factory
2121from configs import cfg_factory
2222from lib .cityscapes_cv2 import get_data_loader
23- from evaluate import eval_model
23+ from tools . evaluate import eval_model
2424from lib .ohem_ce_loss import OhemCELoss
2525from lib .lr_scheduler import WarmupPolyLrScheduler
2626from lib .meters import TimeMeter , AvgMeter
Original file line number Diff line number Diff line change 22# -*- encoding: utf-8 -*-
33
44import sys
5- sys .path .insert (0 , './lib ' )
5+ sys .path .insert (0 , '.' )
66import os
77import os .path as osp
88import random
You can’t perform that action at this time.
0 commit comments