October 24, 2020

222 words 2 mins read

AlphaJia/pytorch-faster-rcnn

AlphaJia/pytorch-faster-rcnn

pytorch based implementation faster rcnn

repo name AlphaJia/pytorch-faster-rcnn
repo link https://github.com/AlphaJia/pytorch-faster-rcnn
homepage
language Python
size (curr.) 27 kB
stars (curr.) 64
created 2020-10-21
license

pytorch-faster-rcnn

pytorch based implementation of faster rcnn framework(Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks)

Framework Structure

backbone

This module includes backbone feature extraction network

config

This module includes config parameters in training period and testing period

  • test_config: specify config parameters in testing period like model_file, image_path_dir, save_dir, etc.
  • train_config: specify config parameters in training period like backbone network, batch_size, image_path_dir, anchor_size, ect.

dataloader

This module inherits pytorch dataloader classes, dataset IO.You can also generate your own dataset dataloader IO and put it in this module

test

This module includes the utils function test(common called unit test, also called UT)

  • anchor_utils_test: some unit testing for utils/anchor_utils.py

utils

This module includes some utilies for image processing, network architectures building, anchor generating, loss function, etc.

  • anchor_utils: some basic function for building anchors
  • im_utils: some basic function for image processing

Continue Updating

comments powered by Disqus