SYMBOL INDEX (634 symbols across 43 files) FILE: MouseLabel.py class LabelMouse (line 5) | class LabelMouse(QLabel): method mouseDoubleClickEvent (line 9) | def mouseDoubleClickEvent(self, event): method mouseMoveEvent (line 12) | def mouseMoveEvent(self): class Label_click_Mouse (line 20) | class Label_click_Mouse(QLabel): method mousePressEvent (line 24) | def mousePressEvent(self, event): FILE: data/regn_mysql.sql type `detection_results` (line 23) | CREATE TABLE `detection_results` ( FILE: detect.py function run (line 33) | def run(weights='yolov5s.pt', # model.pt path(s) function parse_opt (line 195) | def parse_opt(): function main (line 225) | def main(opt): FILE: dialog/rtsp_dialog.py class Ui_Form (line 14) | class Ui_Form(object): method setupUi (line 15) | def setupUi(self, Form): method retranslateUi (line 85) | def retranslateUi(self, Form): FILE: dialog/rtsp_win.py class Window (line 6) | class Window(QWidget, Ui_Form): method __init__ (line 7) | def __init__(self): FILE: hubconf.py function _create (line 11) | def _create(name, pretrained=True, channels=3, classes=80, autoshape=Tru... function custom (line 65) | def custom(path='path/to/model.pt', autoshape=True, verbose=True, device... function yolov5s (line 70) | def yolov5s(pretrained=True, channels=3, classes=80, autoshape=True, ver... function yolov5m (line 75) | def yolov5m(pretrained=True, channels=3, classes=80, autoshape=True, ver... function yolov5l (line 80) | def yolov5l(pretrained=True, channels=3, classes=80, autoshape=True, ver... function yolov5x (line 85) | def yolov5x(pretrained=True, channels=3, classes=80, autoshape=True, ver... function yolov5s6 (line 90) | def yolov5s6(pretrained=True, channels=3, classes=80, autoshape=True, ve... function yolov5m6 (line 95) | def yolov5m6(pretrained=True, channels=3, classes=80, autoshape=True, ve... function yolov5l6 (line 100) | def yolov5l6(pretrained=True, channels=3, classes=80, autoshape=True, ve... function yolov5x6 (line 105) | def yolov5x6(pretrained=True, channels=3, classes=80, autoshape=True, ve... FILE: login_lj.py class Login (line 7) | class Login(QtWidgets.QDialog, Ui_Dialog): method __init__ (line 8) | def __init__(self): method login (line 19) | def login(self): method save_credentials (line 36) | def save_credentials(self): method load_credentials (line 45) | def load_credentials(self): FILE: main.py class DetThread (line 67) | class DetThread(QThread): method __init__ (line 83) | def __init__(self): method get_db_connection (line 99) | def get_db_connection(self, db_host=DB_HOST, db_user=DB_USER, db_passw... method insert_detection_result_to_database (line 106) | def insert_detection_result_to_database(self, sign_type, sign_count, a... method run (line 116) | def run(self, class AddRecordDialog (line 367) | class AddRecordDialog(QtWidgets.QDialog): method __init__ (line 368) | def __init__(self, parent=None): method get_record_data (line 387) | def get_record_data(self): class PymysqlTableModel (line 394) | class PymysqlTableModel(QAbstractTableModel): method __init__ (line 395) | def __init__(self, data, headers): method rowCount (line 400) | def rowCount(self, parent=None): method columnCount (line 403) | def columnCount(self, parent=None): method data (line 406) | def data(self, index, role=Qt.DisplayRole): method headerData (line 411) | def headerData(self, section, orientation, role=Qt.DisplayRole): method filter_data (line 415) | def filter_data(self, filter_function): class ImageAugmentor (line 427) | class ImageAugmentor(QObject): method __init__ (line 429) | def __init__(self): method process_images (line 440) | def process_images(self, input_folder: str, output_folder: str, probab... method load_random_image (line 480) | def load_random_image(self, input_folder: str) -> np.ndarray: method apply_operation (line 486) | def apply_operation(self, image: np.ndarray, operation_index: int, method flip (line 502) | def flip(self, image): method random_crop (line 508) | def random_crop(self, image): method random_scale (line 519) | def random_scale(self, image): method random_rotate (line 525) | def random_rotate(self, image): method random_brightness_contrast_saturation_hue (line 533) | def random_brightness_contrast_saturation_hue(self, image): method random_erase (line 551) | def random_erase(self, image): method random_noise (line 563) | def random_noise(self, image): method mixup (line 574) | def mixup(self, image): method cutout (line 588) | def cutout(self, image): method mosaic (line 600) | def mosaic(self, image: np.ndarray, extra_image: np.ndarray) -> np.nda... class MainWindow (line 626) | class MainWindow(QMainWindow, Ui_mainWindow): method __init__ (line 627) | def __init__(self, parent=None): method slider_adjustment (line 838) | def slider_adjustment(self, value): method apply_custom_adjustments_with_value (line 854) | def apply_custom_adjustments_with_value(self): method binarize_image (line 873) | def binarize_image(self, threshold=128, preview=True): method adjust_gamma (line 908) | def adjust_gamma(self, gamma=1.0, preview=True): method adjust_brightness1 (line 939) | def adjust_brightness1(self, value=1.5, preview=True): method adjust_hue (line 978) | def adjust_hue(self, value=1.5, preview=True): method adjust_saturation (line 1015) | def adjust_saturation(self, value=1.5, preview=True): method adjust_contrast (line 1060) | def adjust_contrast(self, alpha=1.5, preview=True): method adjust_brightness (line 1089) | def adjust_brightness(self, beta=50, preview=True): method sharpen_image (line 1112) | def sharpen_image(self, preview=True): method smooth_image (line 1135) | def smooth_image(self, preview=True): method flip_image (line 1157) | def flip_image(self, orientation='水平翻转',preview=True): method histogram_equalization (line 1188) | def histogram_equalization(self, preview=True): method pixmap_to_cv2_image (line 1211) | def pixmap_to_cv2_image(self, pixmap): method cv2_image_to_pixmap (line 1224) | def cv2_image_to_pixmap(self, image): method morphology_operation (line 1232) | def morphology_operation(self, operation, preview=True): method mean_filter (line 1266) | def mean_filter(self, preview=True): method gaussian_filter (line 1288) | def gaussian_filter(self, preview=True): method median_filter (line 1310) | def median_filter(self, preview=True): method load_image1 (line 1332) | def load_image1(self, file_name): method load_image (line 1350) | def load_image(self, file_name): method open_folder (line 1358) | def open_folder(self): method select_save_folder (line 1366) | def select_save_folder(self): method next_image (line 1369) | def next_image(self): method previous_image (line 1377) | def previous_image(self): method save_image (line 1385) | def save_image(self): method apply_image_enhance (line 1406) | def apply_image_enhance(self, preview): method apply_image_filter (line 1423) | def apply_image_filter(self,preview): method apply_morphology (line 1433) | def apply_morphology(self,preview): method apply_histogram (line 1445) | def apply_histogram(self,preview): method scale_image (line 1472) | def scale_image(self): method rotate_image (line 1492) | def rotate_image(self): method show_help (line 1536) | def show_help(self): method checkbox_clicked (line 1606) | def checkbox_clicked(self): method init_progress_bar (line 1621) | def init_progress_bar(self): method update_progress (line 1626) | def update_progress(self, value): method select_input_folder (line 1635) | def select_input_folder(self): method select_output_folder (line 1639) | def select_output_folder(self): method start_processing (line 1643) | def start_processing(self): method open_advanced_settings (line 1663) | def open_advanced_settings(self): method on_visualize_checkbox_stateChanged (line 1701) | def on_visualize_checkbox_stateChanged(self, state): method on_databases_checkbox_stateChanged (line 1705) | def on_databases_checkbox_stateChanged(self, state): method on_agnostic_nmscheckbox_checkbox_stateChanged (line 1709) | def on_agnostic_nmscheckbox_checkbox_stateChanged(self, state): method on_halfcheckbox_2_nmscheckbox_checkbox_stateChanged (line 1713) | def on_halfcheckbox_2_nmscheckbox_checkbox_stateChanged(self, state): method on_save_crop_nmscheckbox_checkbox_stateChanged (line 1717) | def on_save_crop_nmscheckbox_checkbox_stateChanged(self, state): method on_augment_nmscheckbox_checkbox_stateChanged (line 1721) | def on_augment_nmscheckbox_checkbox_stateChanged(self, state): method on_imgsz_spinbox_valueChanged (line 1725) | def on_imgsz_spinbox_valueChanged(self, value): method on_line_thickness_spinbox_valueChanged (line 1729) | def on_line_thickness_spinbox_valueChanged(self, value): method update_classes (line 1733) | def update_classes(self, text): method on_hide_labelscheckbox_checkbox_checkbox_stateChanged (line 1740) | def on_hide_labelscheckbox_checkbox_checkbox_stateChanged(self, state): method on_hide_confcheckbox_checkbox_checkbox_stateChanged (line 1744) | def on_hide_confcheckbox_checkbox_checkbox_stateChanged(self, state): method on_browse_result_folder_button_clicked (line 1748) | def on_browse_result_folder_button_clicked(self): method get_db_connection (line 1764) | def get_db_connection( self, host=DB_HOST, user=DB_USER, password=DB_P... method draw_bar_chart (line 1772) | def draw_bar_chart(self, data): method draw_line_chart (line 1793) | def draw_line_chart(self, data): method draw_pie_chart (line 1814) | def draw_pie_chart(self, data, threshold=5): method draw_box_plot (line 1849) | def draw_box_plot(self, data): method on_analyze_button_clicked (line 1870) | def on_analyze_button_clicked(self): method open_stackedWidget (line 1884) | def open_stackedWidget(self, index): method apply_filter (line 1887) | def apply_filter(self): method filter_data_in_database (line 1892) | def filter_data_in_database(self, user_input): method on_sort_changed (line 1908) | def on_sort_changed(self): method sort_data_in_database (line 1923) | def sort_data_in_database(self, sort_column, qt_sort_order): method on_add_button_clicked (line 1946) | def on_add_button_clicked(self): method insert_data_to_database (line 1955) | def insert_data_to_database(self, sign_type, sign_count, additional_in... method on_delete_button_clicked (line 1966) | def on_delete_button_clicked(self): method fetch_data_from_database (line 1991) | def fetch_data_from_database(self): method refresh_data (line 2009) | def refresh_data(self): method search_pt (line 2024) | def search_pt(self): method is_save (line 2041) | def is_save(self): method checkrate (line 2049) | def checkrate(self): method chose_rtsp (line 2055) | def chose_rtsp(self): method load_rtsp (line 2071) | def load_rtsp(self, ip): method chose_cam (line 2086) | def chose_cam(self): method load_setting (line 2129) | def load_setting(self): method change_val (line 2168) | def change_val(self, x, flag): method statistic_msg (line 2187) | def statistic_msg(self, msg): method show_msg (line 2191) | def show_msg(self, msg): method change_model (line 2197) | def change_model(self, x): method open_file (line 2202) | def open_file(self): method max_or_restore (line 2222) | def max_or_restore(self): method run_or_continue (line 2229) | def run_or_continue(self): method stop (line 2246) | def stop(self): method mousePressEvent (line 2250) | def mousePressEvent(self, event): method mouseMoveEvent (line 2257) | def mouseMoveEvent(self, QMouseEvent): method mouseReleaseEvent (line 2261) | def mouseReleaseEvent(self, QMouseEvent): method show_image (line 2265) | def show_image(img_src, label): method show_statistic (line 2295) | def show_statistic(self, statistic_dic): method closeEvent (line 2306) | def closeEvent(self, event): FILE: main_win/login.py class Ui_Dialog (line 6) | class Ui_Dialog(object): method setupUi (line 7) | def setupUi(self, Dialog): method retranslateUi (line 101) | def retranslateUi(self, Dialog): FILE: main_win/win.py class Ui_mainWindow (line 11) | class Ui_mainWindow(object): method setupUi (line 12) | def setupUi(self, mainWindow): method retranslateUi (line 3994) | def retranslateUi(self, mainWindow): FILE: models/common.py function autopad (line 31) | def autopad(k, p=None): # kernel, padding class Conv (line 38) | class Conv(nn.Module): method __init__ (line 40) | def __init__(self, c1, c2, k=1, s=1, p=None, g=1, act=True): # ch_in,... method forward (line 46) | def forward(self, x): method forward_fuse (line 49) | def forward_fuse(self, x): class DWConv (line 53) | class DWConv(Conv): method __init__ (line 55) | def __init__(self, c1, c2, k=1, s=1, act=True): # ch_in, ch_out, kern... class TransformerLayer (line 59) | class TransformerLayer(nn.Module): method __init__ (line 61) | def __init__(self, c, num_heads): method forward (line 70) | def forward(self, x): class TransformerBlock (line 76) | class TransformerBlock(nn.Module): method __init__ (line 78) | def __init__(self, c1, c2, num_heads, num_layers): method forward (line 87) | def forward(self, x): class Bottleneck (line 95) | class Bottleneck(nn.Module): method __init__ (line 97) | def __init__(self, c1, c2, shortcut=True, g=1, e=0.5): # ch_in, ch_ou... method forward (line 104) | def forward(self, x): class BottleneckCSP (line 108) | class BottleneckCSP(nn.Module): method __init__ (line 110) | def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5): # ch_in, ... method forward (line 121) | def forward(self, x): class C3 (line 127) | class C3(nn.Module): method __init__ (line 129) | def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5): # ch_in, ... method forward (line 138) | def forward(self, x): class C3TR (line 142) | class C3TR(C3): method __init__ (line 144) | def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5): class C3SPP (line 150) | class C3SPP(C3): method __init__ (line 152) | def __init__(self, c1, c2, k=(5, 9, 13), n=1, shortcut=True, g=1, e=0.5): class C3Ghost (line 158) | class C3Ghost(C3): method __init__ (line 160) | def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5): class SPP (line 166) | class SPP(nn.Module): method __init__ (line 168) | def __init__(self, c1, c2, k=(5, 9, 13)): method forward (line 175) | def forward(self, x): class SPPF (line 182) | class SPPF(nn.Module): method __init__ (line 184) | def __init__(self, c1, c2, k=5): # equivalent to SPP(k=(5, 9, 13)) method forward (line 191) | def forward(self, x): class Focus (line 200) | class Focus(nn.Module): method __init__ (line 202) | def __init__(self, c1, c2, k=1, s=1, p=None, g=1, act=True): # ch_in,... method forward (line 207) | def forward(self, x): # x(b,c,w,h) -> y(b,4c,w/2,h/2) class GhostConv (line 212) | class GhostConv(nn.Module): method __init__ (line 214) | def __init__(self, c1, c2, k=1, s=1, g=1, act=True): # ch_in, ch_out,... method forward (line 220) | def forward(self, x): class GhostBottleneck (line 225) | class GhostBottleneck(nn.Module): method __init__ (line 227) | def __init__(self, c1, c2, k=3, s=1): # ch_in, ch_out, kernel, stride method forward (line 236) | def forward(self, x): class Contract (line 240) | class Contract(nn.Module): method __init__ (line 242) | def __init__(self, gain=2): method forward (line 246) | def forward(self, x): class Expand (line 254) | class Expand(nn.Module): method __init__ (line 256) | def __init__(self, gain=2): method forward (line 260) | def forward(self, x): class Concat (line 268) | class Concat(nn.Module): method __init__ (line 270) | def __init__(self, dimension=1): method forward (line 274) | def forward(self, x): class DetectMultiBackend (line 278) | class DetectMultiBackend(nn.Module): method __init__ (line 280) | def __init__(self, weights='yolov5s.pt', device=None, dnn=False, data=... method forward (line 398) | def forward(self, im, augment=False, visualize=False, val=False): method warmup (line 458) | def warmup(self, imgsz=(1, 3, 640, 640), half=False): method model_type (line 466) | def model_type(p='path/to/model.pt'): class AutoShape (line 478) | class AutoShape(nn.Module): method __init__ (line 488) | def __init__(self, model): method _apply (line 496) | def _apply(self, fn): method forward (line 508) | def forward(self, imgs, size=640, augment=False, profile=False): class Detections (line 566) | class Detections: method __init__ (line 568) | def __init__(self, imgs, pred, files, times=(0, 0, 0, 0), names=None, ... method display (line 585) | def display(self, pprint=False, show=False, save=False, crop=False, re... method print (line 624) | def print(self): method show (line 629) | def show(self): method save (line 632) | def save(self, save_dir='runs/detect/exp'): method crop (line 636) | def crop(self, save=True, save_dir='runs/detect/exp'): method render (line 640) | def render(self): method pandas (line 644) | def pandas(self): method tolist (line 654) | def tolist(self): method __len__ (line 663) | def __len__(self): class Classify (line 667) | class Classify(nn.Module): method __init__ (line 669) | def __init__(self, c1, c2, k=1, s=1, p=None, g=1): # ch_in, ch_out, k... method forward (line 675) | def forward(self, x): FILE: models/experimental.py class CrossConv (line 15) | class CrossConv(nn.Module): method __init__ (line 17) | def __init__(self, c1, c2, k=3, s=1, g=1, e=1.0, shortcut=False): method forward (line 25) | def forward(self, x): class Sum (line 29) | class Sum(nn.Module): method __init__ (line 31) | def __init__(self, n, weight=False): # n: number of inputs method forward (line 38) | def forward(self, x): class MixConv2d (line 50) | class MixConv2d(nn.Module): method __init__ (line 52) | def __init__(self, c1, c2, k=(1, 3), s=1, equal_ch=True): # ch_in, ch... method forward (line 71) | def forward(self, x): class Ensemble (line 75) | class Ensemble(nn.ModuleList): method __init__ (line 77) | def __init__(self): method forward (line 80) | def forward(self, x, augment=False, profile=False, visualize=False): function attempt_load (line 90) | def attempt_load(weights, map_location=None, inplace=True, fuse=True): FILE: models/tf.py class TFBN (line 37) | class TFBN(keras.layers.Layer): method __init__ (line 39) | def __init__(self, w=None): method call (line 48) | def call(self, inputs): class TFPad (line 52) | class TFPad(keras.layers.Layer): method __init__ (line 53) | def __init__(self, pad): method call (line 57) | def call(self, inputs): class TFConv (line 61) | class TFConv(keras.layers.Layer): method __init__ (line 63) | def __init__(self, c1, c2, k=1, s=1, p=None, g=1, act=True, w=None): method call (line 88) | def call(self, inputs): class TFFocus (line 92) | class TFFocus(keras.layers.Layer): method __init__ (line 94) | def __init__(self, c1, c2, k=1, s=1, p=None, g=1, act=True, w=None): method call (line 99) | def call(self, inputs): # x(b,w,h,c) -> y(b,w/2,h/2,4c) class TFBottleneck (line 107) | class TFBottleneck(keras.layers.Layer): method __init__ (line 109) | def __init__(self, c1, c2, shortcut=True, g=1, e=0.5, w=None): # ch_i... method call (line 116) | def call(self, inputs): class TFConv2d (line 120) | class TFConv2d(keras.layers.Layer): method __init__ (line 122) | def __init__(self, c1, c2, k, s=1, g=1, bias=True, w=None): method call (line 130) | def call(self, inputs): class TFBottleneckCSP (line 134) | class TFBottleneckCSP(keras.layers.Layer): method __init__ (line 136) | def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5, w=None): method call (line 148) | def call(self, inputs): class TFC3 (line 154) | class TFC3(keras.layers.Layer): method __init__ (line 156) | def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5, w=None): method call (line 165) | def call(self, inputs): class TFSPP (line 169) | class TFSPP(keras.layers.Layer): method __init__ (line 171) | def __init__(self, c1, c2, k=(5, 9, 13), w=None): method call (line 178) | def call(self, inputs): class TFSPPF (line 183) | class TFSPPF(keras.layers.Layer): method __init__ (line 185) | def __init__(self, c1, c2, k=5, w=None): method call (line 192) | def call(self, inputs): class TFDetect (line 199) | class TFDetect(keras.layers.Layer): method __init__ (line 200) | def __init__(self, nc=80, anchors=(), ch=(), imgsz=(640, 640), w=None)... method call (line 218) | def call(self, inputs): method _make_grid (line 240) | def _make_grid(nx=20, ny=20): class TFUpsample (line 247) | class TFUpsample(keras.layers.Layer): method __init__ (line 248) | def __init__(self, size, scale_factor, mode, w=None): # warning: all ... method call (line 257) | def call(self, inputs): class TFConcat (line 261) | class TFConcat(keras.layers.Layer): method __init__ (line 262) | def __init__(self, dimension=1, w=None): method call (line 267) | def call(self, inputs): function parse_model (line 271) | def parse_model(d, ch, model, imgsz): # model_dict, input_channels(3) class TFModel (line 323) | class TFModel: method __init__ (line 324) | def __init__(self, cfg='yolov5s.yaml', ch=3, nc=None, model=None, imgs... method predict (line 340) | def predict(self, inputs, tf_nms=False, agnostic_nms=False, topk_per_c... method _xywh2xyxy (line 374) | def _xywh2xyxy(xywh): class AgnosticNMS (line 380) | class AgnosticNMS(keras.layers.Layer): method call (line 382) | def call(self, input, topk_all, iou_thres, conf_thres): method _nms (line 389) | def _nms(x, topk_all=100, iou_thres=0.45, conf_thres=0.25): # agnosti... function representative_dataset_gen (line 411) | def representative_dataset_gen(dataset, ncalib=100): function run (line 422) | def run(weights=ROOT / 'yolov5s.pt', # weights path function parse_opt (line 446) | def parse_opt(): function main (line 458) | def main(opt): FILE: models/yolo.py class Detect (line 33) | class Detect(nn.Module): method __init__ (line 37) | def __init__(self, nc=80, anchors=(), ch=(), inplace=True): # detecti... method forward (line 49) | def forward(self, x): method _make_grid (line 72) | def _make_grid(self, nx=20, ny=20, i=0): class Model (line 84) | class Model(nn.Module): method __init__ (line 85) | def __init__(self, cfg='yolov5s.yaml', ch=3, nc=None, anchors=None): ... method forward (line 123) | def forward(self, x, augment=False, profile=False, visualize=False): method _forward_augment (line 128) | def _forward_augment(self, x): method _forward_once (line 142) | def _forward_once(self, x, profile=False, visualize=False): method _descale_pred (line 155) | def _descale_pred(self, p, flips, scale, img_size): method _clip_augmented (line 172) | def _clip_augmented(self, y): method _profile_one_layer (line 183) | def _profile_one_layer(self, m, x, dt): method _initialize_biases (line 196) | def _initialize_biases(self, cf=None): # initialize biases into Detec... method _print_biases (line 206) | def _print_biases(self): method fuse (line 218) | def fuse(self): # fuse model Conv2d() + BatchNorm2d() layers method info (line 228) | def info(self, verbose=False, img_size=640): # print model information method _apply (line 231) | def _apply(self, fn): function parse_model (line 243) | def parse_model(d, ch): # model_dict, input_channels(3) FILE: utils/CustomMessageBox.py class MessageBox (line 7) | class MessageBox(QMessageBox): method __init__ (line 8) | def __init__(self, *args, title='提示', count=1, time=1000, auto=False, ... method doCountDown (line 34) | def doCountDown(self): FILE: utils/__init__.py function notebook_init (line 7) | def notebook_init(verbose=True): FILE: utils/activations.py class SiLU (line 12) | class SiLU(nn.Module): # export-friendly version of nn.SiLU() method forward (line 14) | def forward(x): class Hardswish (line 18) | class Hardswish(nn.Module): # export-friendly version of nn.Hardswish() method forward (line 20) | def forward(x): class Mish (line 26) | class Mish(nn.Module): method forward (line 28) | def forward(x): class MemoryEfficientMish (line 32) | class MemoryEfficientMish(nn.Module): class F (line 33) | class F(torch.autograd.Function): method forward (line 35) | def forward(ctx, x): method backward (line 40) | def backward(ctx, grad_output): method forward (line 46) | def forward(self, x): class FReLU (line 51) | class FReLU(nn.Module): method __init__ (line 52) | def __init__(self, c1, k=3): # ch_in, kernel method forward (line 57) | def forward(self, x): class AconC (line 62) | class AconC(nn.Module): method __init__ (line 68) | def __init__(self, c1): method forward (line 74) | def forward(self, x): class MetaAconC (line 79) | class MetaAconC(nn.Module): method __init__ (line 85) | def __init__(self, c1, k=1, s=1, r=16): # ch_in, kernel, stride, r method forward (line 95) | def forward(self, x): FILE: utils/augmentations.py class Albumentations (line 16) | class Albumentations: method __init__ (line 18) | def __init__(self): method __call__ (line 40) | def __call__(self, im, labels, p=1.0): function augment_hsv (line 47) | def augment_hsv(im, hgain=0.5, sgain=0.5, vgain=0.5): function hist_equalize (line 63) | def hist_equalize(im, clahe=True, bgr=False): function replicate (line 74) | def replicate(im, labels): function letterbox (line 91) | def letterbox(im, new_shape=(640, 640), color=(114, 114, 114), auto=True... function random_perspective (line 124) | def random_perspective(im, targets=(), segments=(), degrees=10, translat... function copy_paste (line 213) | def copy_paste(im, labels, segments, p=0.5): function cutout (line 237) | def cutout(im, labels, p=0.5): function mixup (line 264) | def mixup(im, labels, im2, labels2): function box_candidates (line 272) | def box_candidates(box1, box2, wh_thr=2, ar_thr=100, area_thr=0.1, eps=1... FILE: utils/autoanchor.py function check_anchor_order (line 18) | def check_anchor_order(m): function check_anchors (line 28) | def check_anchors(dataset, model, thr=4.0, imgsz=640): function kmean_anchors (line 65) | def kmean_anchors(dataset='./data/coco128.yaml', n=9, img_size=640, thr=... FILE: utils/autobatch.py function check_train_batch_size (line 16) | def check_train_batch_size(model, imgsz=640): function autobatch (line 22) | def autobatch(model, imgsz=640, fraction=0.9, batch_size=16): FILE: utils/benchmarks.py function run (line 46) | def run(weights=ROOT / 'yolov5s.pt', # weights path function parse_opt (line 75) | def parse_opt(): function main (line 86) | def main(opt): FILE: utils/callbacks.py class Callbacks (line 7) | class Callbacks: method __init__ (line 12) | def __init__(self): method register_action (line 40) | def register_action(self, hook, name='', callback=None): method get_registered_actions (line 53) | def get_registered_actions(self, hook=None): method run (line 65) | def run(self, hook, *args, **kwargs): FILE: utils/capnums.py class Camera (line 4) | class Camera: method __init__ (line 5) | def __init__(self, cam_preset_num=5): method get_cam_num (line 8) | def get_cam_num(self): FILE: utils/datasets.py function get_hash (line 45) | def get_hash(paths): function exif_size (line 53) | def exif_size(img): function exif_transpose (line 68) | def exif_transpose(image): function create_dataloader (line 94) | def create_dataloader(path, imgsz, batch_size, stride, single_cls=False,... class InfiniteDataLoader (line 125) | class InfiniteDataLoader(dataloader.DataLoader): method __init__ (line 131) | def __init__(self, *args, **kwargs): method __len__ (line 136) | def __len__(self): method __iter__ (line 139) | def __iter__(self): class _RepeatSampler (line 144) | class _RepeatSampler: method __init__ (line 151) | def __init__(self, sampler): method __iter__ (line 154) | def __iter__(self): class LoadImages (line 159) | class LoadImages: method __init__ (line 161) | def __init__(self, path, img_size=640, stride=32, auto=True): method __iter__ (line 190) | def __iter__(self): method __next__ (line 194) | def __next__(self): method new_video (line 232) | def new_video(self, path): method __len__ (line 237) | def __len__(self): class LoadWebcam (line 241) | class LoadWebcam: # for inference method __init__ (line 243) | def __init__(self, pipe='0', img_size=640, stride=32): method __iter__ (line 250) | def __iter__(self): method __next__ (line 254) | def __next__(self): method __len__ (line 279) | def __len__(self): class LoadStreams (line 283) | class LoadStreams: method __init__ (line 285) | def __init__(self, sources='streams.txt', img_size=640, stride=32, aut... method update (line 328) | def update(self, i, cap, stream): method __iter__ (line 345) | def __iter__(self): method __next__ (line 349) | def __next__(self): method __len__ (line 368) | def __len__(self): function img2label_paths (line 372) | def img2label_paths(img_paths): class LoadImagesAndLabels (line 378) | class LoadImagesAndLabels(Dataset): method __init__ (line 382) | def __init__(self, path, img_size=640, batch_size=16, augment=False, h... method cache_labels (line 509) | def cache_labels(self, path=Path('./labels.cache'), prefix=''): method __len__ (line 545) | def __len__(self): method __getitem__ (line 554) | def __getitem__(self, index): method load_image (line 627) | def load_image(self, i): method load_mosaic (line 648) | def load_mosaic(self, index): method load_mosaic9 (line 704) | def load_mosaic9(self, index): method collate_fn (line 779) | def collate_fn(batch): method collate_fn4 (line 786) | def collate_fn4(batch): function create_folder (line 813) | def create_folder(path='./new'): function flatten_recursive (line 820) | def flatten_recursive(path=DATASETS_DIR / 'coco128'): function extract_boxes (line 828) | def extract_boxes(path=DATASETS_DIR / 'coco128'): # from utils.datasets... function autosplit (line 862) | def autosplit(path=DATASETS_DIR / 'coco128/images', weights=(0.9, 0.1, 0... function verify_image_label (line 886) | def verify_image_label(args): function dataset_stats (line 938) | def dataset_stats(path='coco128.yaml', autodownload=False, verbose=False... FILE: utils/downloads.py function gsutil_getsize (line 18) | def gsutil_getsize(url=''): function safe_download (line 24) | def safe_download(file, url, url2=None, min_bytes=1E0, error_msg=''): function attempt_download (line 43) | def attempt_download(file, repo='ultralytics/yolov5'): # from utils.dow... function gdrive_download (line 83) | def gdrive_download(id='16TiPfZj7htmTyhntwcZyEEAejOUxuT6m', file='tmp.zi... function get_token (line 118) | def get_token(cookie="./cookie"): FILE: utils/flask_rest_api/restapi.py function predict (line 17) | def predict(): FILE: utils/general.py function is_kaggle (line 50) | def is_kaggle(): function is_writeable (line 60) | def is_writeable(dir, test=False): function set_logging (line 75) | def set_logging(name=None, verbose=VERBOSE): function user_config_dir (line 88) | def user_config_dir(dir='Ultralytics', env_var='YOLOV5_CONFIG_DIR'): class Profile (line 104) | class Profile(contextlib.ContextDecorator): method __enter__ (line 106) | def __enter__(self): method __exit__ (line 109) | def __exit__(self, type, value, traceback): class Timeout (line 113) | class Timeout(contextlib.ContextDecorator): method __init__ (line 115) | def __init__(self, seconds, *, timeout_msg='', suppress_timeout_errors... method _timeout_handler (line 120) | def _timeout_handler(self, signum, frame): method __enter__ (line 123) | def __enter__(self): method __exit__ (line 127) | def __exit__(self, exc_type, exc_val, exc_tb): class WorkingDirectory (line 133) | class WorkingDirectory(contextlib.ContextDecorator): method __init__ (line 135) | def __init__(self, new_dir): method __enter__ (line 139) | def __enter__(self): method __exit__ (line 142) | def __exit__(self, exc_type, exc_val, exc_tb): function try_except (line 146) | def try_except(func): function methods (line 157) | def methods(instance): function print_args (line 162) | def print_args(name, opt): function init_seeds (line 167) | def init_seeds(seed=0): function intersect_dicts (line 177) | def intersect_dicts(da, db, exclude=()): function get_latest_run (line 182) | def get_latest_run(search_dir='.'): function is_docker (line 188) | def is_docker(): function is_colab (line 193) | def is_colab(): function is_pip (line 202) | def is_pip(): function is_ascii (line 207) | def is_ascii(s=''): function is_chinese (line 213) | def is_chinese(s='人工智能'): function emojis (line 218) | def emojis(str=''): function file_size (line 223) | def file_size(path): function check_online (line 234) | def check_online(): function check_git_status (line 246) | def check_git_status(): function check_python (line 265) | def check_python(minimum='3.6.2'): function check_version (line 270) | def check_version(current='0.0.0', minimum='0.0.0', name='version ', pin... function check_requirements (line 283) | def check_requirements(requirements=ROOT / 'requirements.txt', exclude=(... function check_img_size (line 319) | def check_img_size(imgsz, s=32, floor=0): function check_imshow (line 330) | def check_imshow(): function check_suffix (line 345) | def check_suffix(file='yolov5s.pt', suffix=('.pt',), msg=''): function check_yaml (line 356) | def check_yaml(file, suffix=('.yaml', '.yml')): function check_file (line 361) | def check_file(file, suffix=''): function check_font (line 386) | def check_font(font=FONT): function check_dataset (line 395) | def check_dataset(data, autodownload=True): function url2file (line 450) | def url2file(url): function download (line 457) | def download(url, dir='.', unzip=True, delete=True, curl=False, threads=1): function make_divisible (line 491) | def make_divisible(x, divisor): function clean_str (line 498) | def clean_str(s): function one_cycle (line 503) | def one_cycle(y1=0.0, y2=1.0, steps=100): function colorstr (line 508) | def colorstr(*input): function labels_to_class_weights (line 533) | def labels_to_class_weights(labels, nc=80): function labels_to_image_weights (line 552) | def labels_to_image_weights(labels, nc=80, class_weights=np.ones(80)): function coco80_to_coco91_class (line 560) | def coco80_to_coco91_class(): # converts 80-index (val2014) to 91-index... function xyxy2xywh (line 572) | def xyxy2xywh(x): function xywh2xyxy (line 582) | def xywh2xyxy(x): function xywhn2xyxy (line 592) | def xywhn2xyxy(x, w=640, h=640, padw=0, padh=0): function xyxy2xywhn (line 602) | def xyxy2xywhn(x, w=640, h=640, clip=False, eps=0.0): function xyn2xy (line 614) | def xyn2xy(x, w=640, h=640, padw=0, padh=0): function segment2box (line 622) | def segment2box(segment, width=640, height=640): function segments2boxes (line 630) | def segments2boxes(segments): function resample_segments (line 639) | def resample_segments(segments, n=1000): function scale_coords (line 648) | def scale_coords(img1_shape, coords, img0_shape, ratio_pad=None): function clip_coords (line 664) | def clip_coords(boxes, shape): function non_max_suppression (line 676) | def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, cla... function strip_optimizer (line 770) | def strip_optimizer(f='best.pt', s=''): # from utils.general import *; ... function print_mutation (line 786) | def print_mutation(results, hyp, save_dir, bucket, prefix=colorstr('evol... function apply_classifier (line 828) | def apply_classifier(x, model, img, im0): function increment_path (line 864) | def increment_path(path, exist_ok=False, sep='', mkdir=False): FILE: utils/google_utils.py function gsutil_getsize (line 14) | def gsutil_getsize(url=''): function safe_download (line 20) | def safe_download(file, url, url2=None, min_bytes=1E0, error_msg=''): function attempt_download (line 39) | def attempt_download(file, repo='ultralytics/yolov5'): # from utils.goo... function gdrive_download (line 76) | def gdrive_download(id='16TiPfZj7htmTyhntwcZyEEAejOUxuT6m', file='tmp.zi... function get_token (line 111) | def get_token(cookie="./cookie"): FILE: utils/loggers/__init__.py class Loggers (line 37) | class Loggers(): method __init__ (line 39) | def __init__(self, save_dir=None, weights=None, opt=None, hyp=None, lo... method on_pretrain_routine_end (line 77) | def on_pretrain_routine_end(self): method on_train_batch_end (line 83) | def on_train_batch_end(self, ni, model, imgs, targets, paths, plots, s... method on_train_epoch_end (line 98) | def on_train_epoch_end(self, epoch): method on_val_image_end (line 103) | def on_val_image_end(self, pred, predn, path, names, im): method on_val_end (line 108) | def on_val_end(self): method on_fit_epoch_end (line 114) | def on_fit_epoch_end(self, vals, epoch, best_fitness, fi): method on_model_save (line 136) | def on_model_save(self, last, epoch, final_epoch, best_fitness, fi): method on_train_end (line 142) | def on_train_end(self, last, best, plots, epoch, results): method on_params_update (line 164) | def on_params_update(self, params): FILE: utils/loggers/wandb/log_dataset.py function create_dataset_artifact (line 10) | def create_dataset_artifact(opt): FILE: utils/loggers/wandb/sweep.py function sweep (line 17) | def sweep(): FILE: utils/loggers/wandb/wandb_utils.py function remove_prefix (line 32) | def remove_prefix(from_string, prefix=WANDB_ARTIFACT_PREFIX): function check_wandb_config_file (line 36) | def check_wandb_config_file(data_config_file): function check_wandb_dataset (line 43) | def check_wandb_dataset(data_file): function get_run_info (line 59) | def get_run_info(run_path): function check_wandb_resume (line 68) | def check_wandb_resume(opt): function process_wandb_config_ddp_mode (line 82) | def process_wandb_config_ddp_mode(opt): class WandbLogger (line 106) | class WandbLogger(): method __init__ (line 120) | def __init__(self, opt, run_id=None, job_type='Training'): method check_and_upload_dataset (line 192) | def check_and_upload_dataset(self, opt): method setup_training (line 210) | def setup_training(self, opt): method download_dataset_artifact (line 262) | def download_dataset_artifact(self, path, alias): method download_model_artifact (line 282) | def download_model_artifact(self, opt): method log_model (line 300) | def log_model(self, path, opt, epoch, fitness_score, best_model=False): method log_dataset_artifact (line 324) | def log_dataset_artifact(self, data_file, single_cls, project, overwri... method map_val_table_path (line 381) | def map_val_table_path(self): method create_dataset_table (line 391) | def create_dataset_table(self, dataset: LoadImagesAndLabels, class_to_... method log_training_progress (line 433) | def log_training_progress(self, predn, path, names): method val_one_image (line 474) | def val_one_image(self, pred, predn, path, names, im): method log (line 496) | def log(self, log_dict): method end_epoch (line 507) | def end_epoch(self, best_result=False): method finish_run (line 539) | def finish_run(self): function all_logging_disabled (line 551) | def all_logging_disabled(highest_level=logging.CRITICAL): FILE: utils/loss.py function smooth_BCE (line 13) | def smooth_BCE(eps=0.1): # https://github.com/ultralytics/yolov3/issues... class BCEBlurWithLogitsLoss (line 18) | class BCEBlurWithLogitsLoss(nn.Module): method __init__ (line 20) | def __init__(self, alpha=0.05): method forward (line 25) | def forward(self, pred, true): class FocalLoss (line 35) | class FocalLoss(nn.Module): method __init__ (line 37) | def __init__(self, loss_fcn, gamma=1.5, alpha=0.25): method forward (line 45) | def forward(self, pred, true): class QFocalLoss (line 65) | class QFocalLoss(nn.Module): method __init__ (line 67) | def __init__(self, loss_fcn, gamma=1.5, alpha=0.25): method forward (line 75) | def forward(self, pred, true): class ComputeLoss (line 91) | class ComputeLoss: method __init__ (line 93) | def __init__(self, model, autobalance=False): method __call__ (line 117) | def __call__(self, p, targets): # predictions, targets, model method build_targets (line 169) | def build_targets(self, p, targets): FILE: utils/metrics.py function fitness (line 15) | def fitness(x): function ap_per_class (line 21) | def ap_per_class(tp, conf, pred_cls, target_cls, plot=False, save_dir='.... function compute_ap (line 89) | def compute_ap(recall, precision): class ConfusionMatrix (line 117) | class ConfusionMatrix: method __init__ (line 119) | def __init__(self, nc, conf=0.25, iou_thres=0.45): method process_batch (line 125) | def process_batch(self, detections, labels): method matrix (line 165) | def matrix(self): method tp_fp (line 168) | def tp_fp(self): method plot (line 174) | def plot(self, normalize=True, save_dir='', names=()): method print (line 197) | def print(self): function bbox_iou (line 202) | def bbox_iou(box1, box2, x1y1x2y2=True, GIoU=False, DIoU=False, CIoU=Fal... function box_iou (line 244) | def box_iou(box1, box2): function bbox_ioa (line 269) | def bbox_ioa(box1, box2, eps=1E-7): function wh_iou (line 293) | def wh_iou(wh1, wh2): function plot_pr_curve (line 303) | def plot_pr_curve(px, py, ap, save_dir='pr_curve.png', names=()): function plot_mc_curve (line 324) | def plot_mc_curve(px, py, save_dir='mc_curve.png', names=(), xlabel='Con... FILE: utils/plots.py class Colors (line 30) | class Colors: method __init__ (line 32) | def __init__(self): method __call__ (line 39) | def __call__(self, i, bgr=False): method hex2rgb (line 44) | def hex2rgb(h): # rgb order (PIL) function check_pil_font (line 51) | def check_pil_font(font=FONT, size=10): class Annotator (line 65) | class Annotator: method __init__ (line 70) | def __init__(self, im, line_width=None, font_size=None, font='Arial.tt... method box_label (line 82) | def box_label(self, box, label='', color=(128, 128, 128), txt_color=(2... method rectangle (line 107) | def rectangle(self, xy, fill=None, outline=None, width=1): method text (line 111) | def text(self, xy, text, txt_color=(255, 255, 255)): method result (line 116) | def result(self): function feature_visualization (line 121) | def feature_visualization(x, module_type, stage, n=32, save_dir=Path('ru... function hist2d (line 149) | def hist2d(x, y, n=100): function butter_lowpass_filtfilt (line 158) | def butter_lowpass_filtfilt(data, cutoff=1500, fs=50000, order=5): function output_to_target (line 171) | def output_to_target(output): function plot_images (line 180) | def plot_images(images, targets, paths=None, fname='images.jpg', names=N... function plot_lr_scheduler (line 241) | def plot_lr_scheduler(optimizer, scheduler, epochs=300, save_dir=''): function plot_val_txt (line 258) | def plot_val_txt(): # from utils.plots import *; plot_val() function plot_targets_txt (line 275) | def plot_targets_txt(): # from utils.plots import *; plot_targets_txt() function plot_val_study (line 288) | def plot_val_study(file='', dir='', x=None): # from utils.plots import ... function plot_labels (line 327) | def plot_labels(labels, names=(), save_dir=Path('')): function plot_evolve (line 374) | def plot_evolve(evolve_csv='path/to/evolve.csv'): # from utils.plots im... function plot_results (line 401) | def plot_results(file='path/to/results.csv', dir=''): function profile_idetection (line 427) | def profile_idetection(start=0, stop=0, labels=(), save_dir=''): function save_one_box (line 458) | def save_one_box(xyxy, im, file='image.jpg', gain=1.02, pad=10, square=F... FILE: utils/torch_utils.py function torch_distributed_zero_first (line 34) | def torch_distributed_zero_first(local_rank: int): function date_modified (line 45) | def date_modified(path=__file__): function git_describe (line 51) | def git_describe(path=Path(__file__).parent): # path must be a directory function device_count (line 60) | def device_count(): function select_device (line 70) | def select_device(device='', batch_size=0, newline=True): function time_sync (line 101) | def time_sync(): function profile (line 108) | def profile(input, ops, n=10, device=None): function is_parallel (line 160) | def is_parallel(model): function de_parallel (line 165) | def de_parallel(model): function initialize_weights (line 170) | def initialize_weights(model): function find_modules (line 182) | def find_modules(model, mclass=nn.Conv2d): function sparsity (line 187) | def sparsity(model): function prune (line 196) | def prune(model, amount=0.3): function fuse_conv_and_bn (line 207) | def fuse_conv_and_bn(conv, bn): function model_info (line 230) | def model_info(model, verbose=False, img_size=640): function scale_img (line 254) | def scale_img(img, ratio=1.0, same_shape=False, gs=32): # img(16,3,256,... function copy_attr (line 267) | def copy_attr(a, b, include=(), exclude=()): class EarlyStopping (line 276) | class EarlyStopping: method __init__ (line 278) | def __init__(self, patience=30): method __call__ (line 284) | def __call__(self, epoch, fitness): class ModelEMA (line 299) | class ModelEMA: method __init__ (line 305) | def __init__(self, model, decay=0.9999, updates=0): method update (line 315) | def update(self, model): method update_attr (line 327) | def update_attr(self, model, include=(), exclude=('process_group', 're... FILE: utils/tt100k_to_voc-main/1_build_voc_dir.py function make_voc_dir (line 5) | def make_voc_dir(): FILE: utils/tt100k_to_voc-main/2_json2xml.py function edit_xml (line 6) | def edit_xml(objects, id, dir): function getDirId (line 59) | def getDirId(dir): # get the id list of id.png function is_tt45 (line 71) | def is_tt45(objects): FILE: utils/tt100k_to_voc-main/3_delete_jpg_and_xml.py function delete_train_jpg (line 5) | def delete_train_jpg(train_txt): function delete_test_jpg (line 15) | def delete_test_jpg(test_txt): function delete_train_xml (line 24) | def delete_train_xml(train_txt): function delete_test_xml (line 34) | def delete_test_xml(test_txt): FILE: utils/tt100k_to_voc-main/5_label.py function convert (line 65) | def convert(size, box): function convert_annotation (line 81) | def convert_annotation(image_name): FILE: utils/wandb_logging/log_dataset.py function create_dataset_artifact (line 10) | def create_dataset_artifact(opt): FILE: utils/wandb_logging/sweep.py function sweep (line 14) | def sweep(): FILE: utils/wandb_logging/wandb_utils.py function remove_prefix (line 26) | def remove_prefix(from_string, prefix=WANDB_ARTIFACT_PREFIX): function check_wandb_config_file (line 30) | def check_wandb_config_file(data_config_file): function get_run_info (line 37) | def get_run_info(run_path): function check_wandb_resume (line 46) | def check_wandb_resume(opt): function process_wandb_config_ddp_mode (line 60) | def process_wandb_config_ddp_mode(opt): class WandbLogger (line 84) | class WandbLogger(): method __init__ (line 98) | def __init__(self, opt, name, run_id, data_dict, job_type='Training'): method check_and_upload_dataset (line 145) | def check_and_upload_dataset(self, opt): method setup_training (line 155) | def setup_training(self, opt, data_dict): method download_dataset_artifact (line 192) | def download_dataset_artifact(self, path, alias): method download_model_artifact (line 201) | def download_model_artifact(self, opt): method log_model (line 213) | def log_model(self, path, opt, epoch, fitness_score, best_model=False): method log_dataset_artifact (line 227) | def log_dataset_artifact(self, data_file, single_cls, project, overwri... method map_val_table_path (line 258) | def map_val_table_path(self): method create_dataset_table (line 264) | def create_dataset_table(self, dataset, class_to_id, name='dataset'): method log_training_progress (line 295) | def log_training_progress(self, predn, path, names): method val_one_image (line 317) | def val_one_image(self, pred, predn, path, names, im): method log (line 332) | def log(self, log_dict): method end_epoch (line 337) | def end_epoch(self, best_result=False): method finish_run (line 354) | def finish_run(self): function all_logging_disabled (line 363) | def all_logging_disabled(highest_level=logging.CRITICAL):