Skip to content

gustavz/yolo_for_tf_od_api

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This is a collection of all necessary files to include YoloV2 / Darknet to tensorflow's object_detection repo

Note: The original files in the tensorflow/models/research/object_detection repo changed since this project (March 2018). Manual copying of the yolo related functions into the updated files of the repo may be necessary!

Added or Updated file list

These files need to be copied into the following tensorflow/models/ directories

Hint: To be secure you may want to backup the original files that will be overwritten (although the new files do not change/delete existing code but only extend it)

Config:
    research/object_detection/samples/configs/yolo_v2_darknet_19_voc.config

Anchor box:
    research/object_detection/anchor_generators/yolo_grid_anchor_generator.py
    research/object_detection/anchor_generators/yolo_grid_anchor_generator_test.py
    research/object_detection/builders/anchor_generator_builder.py

Loss:
    research/object_detection/core/losses.py
    research/object_detection/builders/losses_builder.py

Feature map:
    research/slim/nets/darknet.py
    research/slim/nets/darknet_test.py
    research/object_detection/models/yolo_v2_darknet_19_feature_extractor.py
    research/object_detection/models/yolo_v2_darknet_19_feature_extractor_test.py

Proto:
    research/object_detection/protos/anchor_generator.proto
    research/object_detection/protos/yolo_anchor_generator.proto
    research/object_detection/protos/losses.proto

ETC:
    research/object_detection/builders/model_builder.py
    research/object_detection/meta_architectures/yolo_meta_arch.py

©rky0930

About

Files Added or Updated to be able to use yolo-darknet in tensorflows object detection api

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages